1.安装pytorch
# CUDA 10.2
conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.2 -c pytorch
2.安装CUB 构建时间依赖
conda install -c bottler nvidiacub
3.从GitHub 安装,pytorch3d
pip install "git+https://github.com/facebookresearch/pytorch3d.git"
1.安装pytorch
# CUDA 10.2
conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.2 -c pytorch
2.安装CUB 构建时间依赖
conda install -c bottler nvidiacub
3.从GitHub 安装,pytorch3d
pip install "git+https://github.com/facebookresearch/pytorch3d.git"