ubuntu安装pytorch3d

环境

ubuntu 20.04.3 LTS
CUDA 10.1

依赖包安装

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

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

• python 3.8.12
• torchvision 0.8.2
• fvcore 0.1.5
• iopath 0.1.9
• gcc&g++ 7.5.0

因为要自己编译，所以需要下载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是失败的，总是找不到包，所以需要自己编译，参考文章，成功安装。

1. 不要从github下载源码，下载修改过的源码
2. 进入你的虚拟环境，进入pytorch3d的目录，编译
   python setup.py install >yourpath/yourfileName.txt
   yourfileName.txt是用来保存编译过程中的输出日志，以便后面查看编译错误

按照以上步骤，我就成功安装pytorch3d了，编译过程中并没有报错。

关键点

• cuda，cudatoolkit版本要对应
• 编译时，不要用官网的源码，使用文中说的修改过的源码