ubuntu24下安装pytorch3d

是小峰呀2024-09-06 21:43

pytoch3d的安装，首先根据自己torch和cuda的版本信息去下面的网站下载对应的pytroch3d。

bash 复制代码

https://anaconda.org/pytorch3d/pytorch3d/files?page=2

下载到本地之后使用下面的命令进行安装即可

bash 复制代码

conda install pytorch3d-0.7.5-py39_cu121_pyt210.tar.bz2

一般情况下，安装完成之后会缺少两个包，安装上即可。

bash 复制代码

pip install fvcore
pip install iopath

最后，在终端通过下面的命令进行简单测试。

bash 复制代码

python -c "import torch; import pytorch3d; print('PyTorch version:', torch.__version__); print('PyTorch3D version:', pytorch3d.__version__)"

出现上述版本信息，表明安装成功。