Docker安装 nvidia

Docker安装 nvidia

1. 安装Docker

这部分就不说了

2. 配置存储库和密钥

 distribution=$(. /etc/os-release;echo $ID$VERSION_ID) && curl -s -L  | sudo apt-key add - && curl -s -L /$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list

3. 添加源

curl -s -L /$distribution/nvidia-container-runtime.list | sudo tee /etc/apt/sources.list.d/nvidia-container-runtime.list

4. 更新源 安装nvidia-docker

sudo apt-get update
sudo apt-get install -y nvidia-docker2

5.重启

sudo systemctl restart docker

6. 安装镜像

sudo docker run --rm --gpus all nvidia/cuda:11.0-base nvidia-smi