Nvidia-Docker安装、使用基础

一、平台环境

先说一下我自己的平台环境:

  • 系统:Ubuntu 20.04.2 LTS

  • CPU:Intel Core i7-7700 @ 3.60GHz*8

  • 显卡:NVIDIA GeForce RTX 2060

  • 显卡驱动版本:450.102.04

  • CUDA版本:11.0.3

二、Docker的安装和启动

1、运行下面的命令安装:

sudo apt-get update
sudo apt-get install docker.io

2、运行下面的命令启动Docker:

systemctl start docker
systemctl enable docker

3、运行这条命令检查是否安装成功:

sudo docker version

三、安装NVIDIA Container Toolkit

1、添加源(安装之前请先确保显卡驱动已经安装好了)

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

2、安装NVIDIA Container Toolkit并重启Docker

sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit
sudo systemctl restart docker

3、安装完成之后可以新建一个容器测试一下:

sudo docker run -it --name test_nvidia_docker --gpus all nvidia/cuda:11.0-base

其中最后的参数是nvidia/cuda:11.0-base是根据我的cuda版本设置的参数,具体可以根据工作站主机的cuda版本进行修改,版本查看旧用nvcc -V查看。

进入容器之后可以跑一下nvidia-smi查看到显卡的驱动,然后就可以在docker里正常使用GPU了。

四、Docker常见使用方法

4.1 镜像的生成方法:

1、原创:从无到有,创建镜像

2、借用:下载别人创建好的镜像

3、改进:在现有的镜像上创建新的镜像

查看镜像:

sudo docker iamges

查找镜像:

sudo docker search [OPTION] [TERM]

获取镜像:

sudo docker pull [OPTION] NAME[:TAG|@DIGEST]

删除镜像:

sudo docker rmi [OPTION] IMAGE[IMAGE...]

4.2 容器的基本管理和使用

运行一个新容器:

sudo docker run -it --name [NAME] -V [SOURCE:DESTINATION] --gpus all nvidia/cuda:11.0-base

启动一个已有容器:

sudo docker start [NAME]
上一篇:【Pix2Pix】当生成式对抗神经网络遇到车道线检测


下一篇:2021-11-15