1. Create a virtual environment
conda create -n pytorch3d python=3.8
2. Activate the environment
conda activate pytorch3d
3. Install pytorch, note that if you have changed the conda source, you need to add -c pytorch -c conda-forge, otherwise it will be added from the official source by default
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3
4. Install dependent libraries
conda install -c fvcore -c iopath -c conda-forge fvcore iopath
5. Install pytorch3d
conda install pytorch3d -c pytorch3d