原本的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里调用显卡来进行我们的开发了!!!