nvidia-docker容器配置
背景
在服务器上搭建docker容器,并远程ssh连接。在本服务器上不运行代码,可以不安装cuda和cudnn等,保证安装nvidia显卡驱动,就可以在容器中安装cuda调用nvidia显卡驱动。由于可以拉取带有cuda的镜像,因此不需要手动安装cuda等。本文主要讲述在安装好系统和显卡驱动的前提下,进行服务器的建立容器并远程ssh的操作。
前提
已经成功安装linux系统和nvidia显卡驱动
sudo nvidia-smi
安装docker
1、下载安装包
打开终端,输入如下命令:
wget -qO- https://get.docker.com/ | sh
2、更新docker用户组
打开终端,输入如下命令:
sudo groupadd docker
sudo gpasswd -a $XXX docker
sudo gpasswd -a $USER docker
newgrp docker
3、测试安装成功
输入如下命令:
docker run hello-world
输出如下内容,则安装成功:
Hello from Docker!
This message shows that your installation appears to be working correctly.
To generate this message, Docker took the following steps:
1. The Docker client contacted the Docker daemon.
2. The Docker daemon pulled the "hello-world" image from the Docker Hub.
(amd64)
3. The Docker daemon created a new container from that image which runs the
executable that produces the output you are currently reading.
4. The Docker daemon