|
1.首先安装nvidia驱动
2.安装docker-ce(docker.io貌似不可以)
# 1
sudo apt-get update
sudo apt-get -y install apt-transport-https ca-certificates curl software-properties-common
# 2: 安装GPG证书
curl -fsSL http://mirrors.aliyun.com/docker-ce/linux/ubuntu/gpg | sudo apt-key add -
# 3: 写入软件源信息
sudo add-apt-repository "deb [arch=amd64] http://mirrors.aliyun.com/docker-ce/linux/ubuntu $(lsb_release -cs) stable"
# 4: 更新并安装 Docker-CE
sudo apt-get -y update
sudo apt-get -y install docker-ce
3. 安装 nvidia-docker2
# 执行命令 Add the package repositories
# command 1
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \
sudo apt-key add -
# command 2
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
# command 3
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
# 正式安装 Install nvidia-docker2 and reload the Docker daemon configuration
sudo apt-get install -y nvidia-docker2
sudo pkill -SIGHUP dockerd
4.测试
sudo docker run --runtime=nvidia --rm nvidia/cuda:8.0-devel nvidia-smi
sudo nvidia-docker run --rm nvidia/cuda:8.0-devel nvidia-smi
sudo docker run --runtime=nvidia --name cuda -ti -v/:/mnt nvidia/cuda:8.0-devel /bin/bash
或者直接下载ubuntu，然后在container里安装nvidia驱动，cuda+cudnn
ref to https://blog.csdn.net/quantum7/article/details/86416600
https://blog.csdn.net/y19930105/article/details/80763467
https://blog.csdn.net/qq_38079008/article/details/83620573 | 
转播
