ubuntu安装pytorch3d

依赖包安装
按照官网建议安装依赖包，以下是我的版本

conda create -n pytorch3d python=3.8
conda activate pytorch3d
conda install -c pytorch pytorch=1.7.1 torchvision cudatoolkit=10.1
conda install fvcore  iopath

因为要自己编译，所以需要下载CUB。从github上下载CUB-1.10.0的release版本，然后解压缩，创建CUB_HOME环境变量

curl -LO https://github.com/NVIDIA/cub/archive/1.10.0.tar.gz
tar xzf 1.10.0.tar.gz
export CUB_HOME=$PWD/cub-1.10.0

还有一些测试Demo的依赖包

# Demos and examples
conda install jupyter
pip install scikit-image matplotlib imageio plotly opencv-python

# Tests/Linting
pip install black 'isort<5' flake8 flake8-bugbear flake8-comprehensions

编译pytorch3d
我直接在conda安装pytorch3d是失败的，总是找不到包，所以需要自己编译，参考文章，成功安装。

不要从github下载源码，下载修改过的源码（GitHub - yuanluxu/pytorch3d: PyTorch3d is FAIR's library of reusable components for deep learning with 3D data.）
进入你的虚拟环境，进入pytorch3d的目录，编译
python setup.py install >yourpath/yourfileName.txt
yourfileName.txt是用来保存编译过程中的输出日志，以便后面查看编译错误
按照以上步骤，我就成功安装pytorch3d了，编译过程中并没有报错。