[系统运维]Ubuntu20.10下docker使用GPU教程

docker安装过程简略，网上教程很多

关键是docker使用GPU需要安装nvidia**-container-**toolkit，这个在Ubuntu20.10版本下官方是没有支持的，所以会出现这个issue提交的问题Install nvidia-docker on Ubuntu 20.10. · Issue #1407 · NVIDIA/nvidia-docker (github.com)

# Unsupported distribution!
# Check https://nvidia.github.io/nvidia-docker

下面的讨论中指出的问题的解决方案，直接强制将Ubuntu18.04或者Ubuntu20.04的仓库源写入到文件中即可

curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \
  sudo apt-key add -
distribution=$(. /etc/os-release;echo ${ID}20.04)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | \
  sudo tee /etc/apt/sources.list.d/nvidia-docker.list
sudo apt-get update

然后再安装

sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit 

记住还要重启一下docker，sudo systemctl restart docker

最终运行一个cuda测试即可docker run --gpus all nvidia/cuda:9.0-base nvidia-smi