See here to install PyTorch. Example command:
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
To install PyTorch3d, run:
conda install pytorch3d -c pytorch3d
If this causes trouble, building from source works reliably, but you must make sure you have the correct CUDA version on your workstation:
pip install "git+https://github.com/facebookresearch/pytorch3d.git"
See the PyTorch3d installation page for more information.