PyTorch3D 安装

1. PyTorch 安装

• PyTorch 官网：PyTorch 官网

• 最新版本的 PyTorch：最新版本的 PyTorch

以 cuda 12.1，Python 3.10，Pytorch 2.1.2 为例

# conda 安装
conda install pytorch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 pytorch-cuda=12.1 -c pytorch -c nvidia
# pip 安装
pip install torch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/cu121

2. PyTorch3D 安装

• 由于直接使用 pip 进行PyTorch3D 安装会报错，所以选择在官网直接下载对应的安装包进行安装。

• 进入 Anaconda 官网找到 PyTorch3D：PyTorch3D 官网

找到和 cuda 以及 Python 版本对应的 Pytorch3D 版本，找到正确的版本后，右键绿色的文件名选择复制链接地址。暂时没有找到pytorch2.1.2版本，所以选择了pyt210版本。当然，如果没有找到对应你环境的版本，也可以重新创建一个对应的环境，再进行pytorch和pytorch3D的安装即可。

2.1 安装PyTorch3D

使用 conda install + 复制的 PyTorch3D 链接

如 安装对应 Python 3.10 cuda 12.1 Pytorch 2.1.2 版本的 PyTorch3D，使用如下命令：

conda install https://anaconda.org/pytorch3d/pytorch3d/0.7.5/download/linux-64/pytorch3d-0.7.5-py310_cu121_pyt210.tar.bz2

2.2 查看是否安装成功

conda list

成功安装！

参考：
PyTorch3D 安装记录