文章目录
环境
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 op