docker及nvidia-docker安装(深度学习环境配置)

docker安装:

step 1: 安装必要的一些系统工具

sudo apt-get update
sudo apt-get -y install apt-transport-https ca-certificates curl software-properties-common

step 2: 安装GPG证书

curl -fsSL http://mirrors.aliyun.com/docker-ce/linux/ubuntu/gpg | sudo apt-key add -

Step 3: 写入软件源信息

sudo add-apt-repository "deb [arch=amd64] http://mirrors.aliyun.com/docker-ce/linux/ubuntu $(lsb_release -cs) stable"

Step 4: 更新并安装 Docker-CE

sudo apt-get -y update
sudo apt-get -y install docker-ce

Step 5: 重启容器服务:

sudo systemctl daemon-reload
sudo systemctl restart docker

安装NVIDIA-docker

Step 1: 卸载旧版本nvidia-docker(若已安装过,可跳过)

sudo docker volume ls -q -f driver=nvidia-docker | xargs -r -I{} -n1 docker ps -q -a -f volume={} | xargs -r docker rm -f

sudo apt-get purge -y nvidia-docker

sudo apt autoremove

Setp 2: 添加apt-key

curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -

Step 3: 添加系统版本

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)

Step 4: 更新sourcelist

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

Step 5: 安装nvidia-docker

sudo apt-get install -y nvidia-docker2
sudo pkill -SIGHUP dockerd

Step 6: 重启容器

sudo systemctl restart docker

测试demo

sudo docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi

猜你喜欢

转载自blog.csdn.net/weixin_39274106/article/details/127102718