Ubuntu18.04安装docker及nvidia docker 2
之前的时候记得安装完docker之后还需要安装单独的nvidia docker 2,现在的话只需要安装nvidia container toolkit即可
1、docker安装
官网上又详细的介绍
Install Docker Engine on Ubuntudocs.docker.com
或者
sudo apt-get update
sudo apt-get install -y docker.io
systemctl start docker
systemctl enable docker
docker version
2、安装NVIDIA Container Toolkit

官网安装步骤
NVIDIA/nvidia-dockergithub.com
其实就是
##首先要确保已经安装了nvidia driver
# 1、添加源
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
# 2、安装并重启
sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit
sudo systemctl restart docker
# 3、测试
docker run --name test1 -it --gpus all或者写"device=0"0 nvidia/cuda:10.0-base
