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
与具体