使用官方安装脚本自动安装docker

curl -fsSL https://test.docker.com -o test-docker.sh
sudo sh test-docker.sh

安装完后，终端输入docker 即可验证

安装NVIDIA Container Runtime以支持docker调用GPU

添加脚本install_docker_gpu.sh

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

执行脚本

sh install_docker_gpu.sh

安装

apt-get install nvidia-container-runtime

验证

docker run -it --rm --gpus all ubuntu nvidia-smi

到这里，支持调用GPU的docker就安装完成了，加入"–gpus"参数，即可在docker容器内使用GPU资源