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

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

一、docker安装:

Step1:更新系统软件包

sudo apt update

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

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

Step3:添加Docker官方GPG密钥

sudo -i curl -fsSL https://mirrors.aliyun.com/docker-ce/linux/ubuntu/gpg | gpg --dearmor -o /etc/apt/trusted.gpg.d/docker-ce.gpg

Step4:验证

sudo apt-key fingerprint 0EBFCD88

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

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

Step5:再次更新软件包

sudo apt update

Step6:安装默认最新版

sudo apt install docker-ce docker-ce-cli containerd.io

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

sudo vim /etc/docker/daemon.json 

在其中添加

{"registry-mirrors": ["https://alzgoonw.mirror.aliyuncs.com"]}

之后

sudo chomd 777 daemon.json
sudo systemctl daemon-reload
sudo systemctl daemon-reload
sudo systemctl status docker
sudo docker run hello-world

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

sudo docker images

Step9:配置用户组

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

Step10:其他docker运行命令

查看状态:sudo systemctl status docker
启动:sudo systemctl start docker
开机自启:sudo systemctl enable docker
停止:sudo systemctl stop docker

二、加载自己的镜像

sudo docker load -i ubuntu20.04_yolov5.tgz

三、docker加载显卡

sudo curl -s -L https://nvidia.github.io/nvidia-container-runtime/gpgkey | sudo apt-key add -
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
sudo curl -s -L https://nvidia.github.io/nvidia-container-runtime/$distribution/nvidia-container-runtime.list | sudo tee /etc/apt/sources.list.d/nvidia-container-runtime.list
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 www.baidu.com

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

  • 8
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值