1、配置pytorch1.10+cuda11.0

pip install torch==1.10.1+cu111 torchvision==0.11.2+cu111 torchaudio==0.10.1 -f https://download.pytorch.org/whl/cu111/torch_stable.html

2、配置 MinkowskiEngine库

不按下面步骤，出现错误

1、下载MinkowskiEngine==0.5.4到本地

2、查看自己电脑的计算能力，要和库匹配

import torch
print(f"GPU Name: {torch.cuda.get_device_name(0)}")
print(f"Compute Capability: {torch.cuda.get_device_properties(0).major}.{torch.cuda.get_device_properties(0).minor}")

3、安装MinkowskiEngine库

export TORCH_CUDA_ARCH_LIST="7.0"
export CXX=c++; export CUDA_HOME=/usr/local/cuda-11.0; python setup.py install --blas=openblas --force_cuda

因为的计算能力是7.0,cuda=11.0

3、安装 pytorch3d

要自己下载安装包pytorch3d==0.7.4

解压:cd pytorch3d && pip install -e