mmdetection3d框架安装与Demo模型运行--基于Ubuntu18.04+Cuda10.1

1.NVIDIA Driver和Cuda安装

在Ubuntu18.04机器上安装好NVIDIA Driver4.18和CUDA10.1，版本号分别为4.18和10.1

查看NVIDIA Driver版本号：nvidia-smi

查看CUDA版本号：nvcc -V

2.安装MiniConda，并创建和管理虚拟环境

2.1 安装MiniConda，官网：Miniconda — conda documentation

2.2 创建名为openmmlab虚拟环境

conda create --name openmmlab python=3.8 -y

2.3 激活并进入该虚拟环境

conda activate openmmlab

3.安装与CUDA版本号匹配的pytorch，这里安装的是1.8.1

#CUDA 10.1

pip install torch==1.8.1+cu101 torchvision==0.9.1+cu101 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

与具体CUDA版本号对应匹配的pytorch可查询如下页面：