在linux中的docker下搭建cuda环境

显卡驱动安装

根据自己的需要来安装相应的驱动版本

sudo apt install nvidia-drivers-440

安装Docker

sudo apt-get remove docker docker-engine docker.io containerd runc
sudo apt-get update
sudo apt-get install -y apt-transport-https ca-certificates curl gnupg-agent
software-properties-common
curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key
add - sudo apt-key fingerprint 0EBFCD88
sudo add-apt-repository "deb [arch=amd64]
https://download.docker.com/linux/ubuntu $(lsb_release -cs) stable"
sudo apt-get update
sudo apt-get install -y docker-ce docker-ce-cli containerd.io

安装NVIDIA-Docker2

目前使用的是ubuntu16.04，如果使用的是其他版本的Ubuntu，更改第二条命令的ubuntu的版本即可。

curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - 
curl -s -L https://nvidia.github.io/nvidia-docker/ubuntu16.04/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
sudo apt-get update
sudo apt-get install -y nvidia-docker2
sudo pkill -SIGHUP dockerd

测试环境

运行下列代码创建cuda环境

sudo docker run --runtime=nvidia --rm nvidia/cuda:10.2-base
nvidia-smi

如果容器正常创建的话，会出现创建出cuda10.2的容器。如果出现下面的画面，说明环境正常搭建了起来。

如果出现问题的话，考虑是显卡驱动的版本偏低问题，如果不需要这么高版本的cuda，当然也可以来降低cuda版本。