pytoch3d的安装,首先根据自己torch和cuda的版本信息去下面的网站下载对应的pytroch3d。
https://anaconda.org/pytorch3d/pytorch3d/files?page=2
下载到本地之后使用下面的命令进行安装即可
conda install pytorch3d-0.7.5-py39_cu121_pyt210.tar.bz2
一般情况下,安装完成之后会缺少两个包,安装上即可。
pip install fvcore
pip install iopath
最后,在终端通过下面的命令进行简单测试。
python -c "import torch; import pytorch3d; print('PyTorch version:', torch.__version__); print('PyTorch3D version:', pytorch3d.__version__)"
出现上述版本信息,表明安装成功。