通过原yml文件创建好虚拟环境后需要更改下torchvision的版本,更改方式如下,在命令行输入如下命令即可:
pip install torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
然后需要找到pytorch3d的0.5.0版本。然后cd pytorch 直接pip install -e .即可
不换torchvision的话,安装pytorch3d时会报错