服务器中Docker环境搭建配置并使用本机GPU方法

时间：2025-03-29 21:35:42

原本的docker版本在使用gpu时需要安装特定的docker版本，并且需要下载nvidia-docker，但是最新版的docker已经开始支持gpu的调用，具体的方法如下：

一、docker安装：

Step1：更新系统软件包

sudo apt update
  • 1

Step2：安装依赖包【用于通过HTTPS来获取仓库】

sudo apt install apt-transport-https ca-certificates curl software-properties-common
  • 1

Step3：添加Docker官方GPG密钥

sudo -i curl -fsSL /docker-ce/linux/ubuntu/gpg | gpg --dearmor -o /etc/apt//
  • 1

Step4：验证

sudo apt-key fingerprint 0EBFCD88
  • 1

0EBFCD88 是公钥的指纹。执行这个命令后，系统会显示与该指纹相关的公钥信息。
Step4：添加Docker阿里稳定版软件源

sudo add-apt-repository "deb [arch=amd64] /docker-ce/linux/ubuntu $(lsb_release -cs) stable"
  • 1

Step5：再次更新软件包

sudo apt update
  • 1

Step6：安装默认最新版

sudo apt install docker-ce docker-ce-cli
  • 1

Step7：测试，安装好后默认启动新docker环境配置：

sudo vim /etc/docker/
  • 1

在其中添加

{"registry-mirrors": [""]}
  • 1

之后

sudo chomd 777 
sudo systemctl daemon-reload
sudo systemctl daemon-reload
sudo systemctl status docker
sudo docker run hello-world
  • 1
  • 2
  • 3
  • 4
  • 5

如果输出“Hello from Docker!”则表示Docker已经成功安装。
Step8：查看有哪些镜像

sudo docker images
  • 1

Step9：配置用户组

sudo usermod -aG docker galaxfy
  • 1
su - galaxfy  # 刷新shell状态
docker images # 验证
  • 1
  • 2

Step10：其他docker运行命令

查看状态：sudo systemctl status docker
启动：sudo systemctl start docker
开机自启：sudo systemctl enable docker
停止：sudo systemctl stop docker
  • 1
  • 2
  • 3
  • 4

二、加载自己的镜像

sudo docker load -i ubuntu20.04_yolov5.tgz
  • 1

三、docker加载显卡

sudo curl -s -L /nvidia-container-runtime/gpgkey | sudo apt-key add -
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
sudo curl -s -L /nvidia-container-runtime/$distribution/ | sudo tee /etc/apt//
sudo apt-get update
sudo apt-get install nvidia-container-runtime
sudo systemctl restart docker
dockerd --add-runtime=nvidia=/usr/bin/nvidia-container-runtime
sudo docker run -d --rm --gpus all --ipc=host --name yolov5 ab37b1816033 ping
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8

这样，咱们就可以在docker里调用显卡来进行我们的开发了！！！

相关文章