nvidia-docker安装

最新推荐文章于 2024-09-06 13:36:50 发布

SAJIAHAN

最新推荐文章于 2024-09-06 13:36:50 发布

阅读量1.1k

点赞数

文章标签： nvidia-docker docker

本文链接：https://blog.csdn.net/SAJIAHAN/article/details/77503837

版权

一. 安装docker

1. sudo apt-get remove docker docker-engine docker.io
2. sudo apt-get update
3. sudo apt-get install linux-image-extra-$(uname -r) linux-image-extra-virtual
4. sudo apt-get update
5. sudo apt-get install apt-transport-https ca-certificates curl software-properties-common
6. curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -
7. sudo apt-key fingerprint 0EBFCD88
8. lsb_release -cs
9. sudo add-apt-repository "deb [arch=amd64] https://download.docker.com/linux/ubuntu $(lsb_release -cs) stable"
10. sudo apt-get update
11. sudo apt-get install docker-ce

二、配置DNS

1. sudo vim /etc/default/docker
2. 修改文件中的如下配置：
a. DOCKER_OPTS="--registry-mirror=http://hub-mirror.c.163.com --dns 10.167.174.13"
b. export http_proxy="http://IP:PORT/"

三、安装nvidia-docker

1. wget -P /tmp https://github.com/NVIDIA/nvidia-docker/releases/download/v1.0.1/nvidia-docker_1.0.1-1_amd64.deb
2. sudo dpkg -i /tmp/nvidia-docker*.deb && rm /tmp/nvidia-docker*.deb
3. sudo nvidia-docker run --rm nvidia/cuda nvidia-smi