PyTorch3D 安装

1. PyTorch 安装

• PyTorch 官网：PyTorch 官网

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

以 cuda 11.1，Python 3.8，Pytorch 1.8.0 为例

# conda 安装
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge
# pip 安装
pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

2. PyTorch3D 安装

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

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

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

2.1 安装PyTorch3D

使用 conda install + 复制的 PyTorch3D 链接

如 安装对应 Python 3.8 cuda 11.1 Pytorch 1.8.0 版本的 PyTorch3D，使用如下命令：

conda install https://anaconda.org/pytorch3d/pytorch3d/0.7.0/download/linux-64/pytorch3d-0.7.0-py38_cu111_pyt180.tar.bz2

2.2 查看是否安装成功

conda list

成功安装！

参考：
PyTorch3D 安装记录