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__)"

在这里插入图片描述
出现上述版本信息,表明安装成功。

Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐