pytorch3d在linux下安装

首先安装虚拟环境和依赖

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

wget 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

 然后git clone代码并且安装

git clone https://github.com/facebookresearch/pytorch3d.git
cd pytorch3d && pip install -e .

在python里import pytorch3d即可