docker容器使用GPU方法实现

更新时间：2022年05月25日 14:38:18 作者：m0_54850467

本文主要介绍了docker容器使用GPU方法实现，文中通过示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下

1、无nvidia-docker

在早期的时候，没有nvidia-docker，可以通过在容器内再部署一遍nvidia GPU驱动解决。同理，其他设备如果想在容器里使用，也可以采用在容器里重新安装一遍驱动解决。

2、nvidia-docker1.0

nvidia-docker是英伟达公司专门用来为docker容器使用nvidia GPU而设计的，设计方案就是把宿主机的GPU驱动文件映射到容器内部使用，可以通过tensorflow生成GPU驱动文件夹。

3、nvidia-docker2.0

nvidia-docker2.0对nvidia-docker1.0进行了很大的优化，不用再映射宿主机GPU驱动了，直接把宿主机的GPU运行时映射到容器即可。启动方式示例：

nvidia-docker run -d -e NVIDIA_VISIBLE_DEVICES=all --name nvidia_docker_test  nvidia/cuda:10.0-base /bin/sh -c "while true; do echo hello world; sleep 1; done"

4、安装docker-19.03及以上版本

docker19.03及以上版本，已经内置了nvidia-docker，无需再单独部署nvidia-docker了。安装方式如下：

安装docker：

yum install -y yum-utils
yum-config-manager     --add-repo     https://download.docker.com/linux/centos/docker-ce.repo
yum-config-manager --enable docker-ce-nightly
yum-config-manager --enable docker-ce-test
yum install docker-ce docker-ce-cli containerd.io
systemctl start docker

安装nvidia-container-toolkit

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.repo | sudo tee /etc/yum.repos.d/nvidia-docker.repo

sudo yum install -y nvidia-container-toolkit
sudo systemctl restart docker

启动容器：

docker run --gpus all nvidia/cuda:10.0-base /bin/sh -c "while true; do echo hello world; sleep 1; done"

进入容器并输入nvidia-smi验证。

到此这篇关于docker容器使用GPU方法实现的文章就介绍到这了,更多相关docker使用GPU内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

docker
GPU

Docker发布镜像到DockerHub与阿里云容器
本文主要介绍了Docker发布镜像到DockerHub与阿里云容器，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2022-07-07
docker 使用GPU的过程详解
本文以TensorFlow2.0为例给大家介绍docker 使用GPU的过程详解，文中给大家介绍了基于拉的tf-gpu镜像构建自己的镜像的操作方法，启动镜像检查GPU是否可用的相关知识，感兴趣的朋友跟随小编一起看看吧
2021-11-11
远程连接Docker上的Mysql失败的分析与解决方案
这篇文章主要介绍了远程连接Docker上的Mysql失败的分析与解决方案的相关资料,需要的朋友可以参考下
2023-01-01
使用docker部署influxdb与 mongo的常用命令
这篇文章主要介绍了使用docker部署influxdb与 mongo的常用命令,本文给大家介绍的非常详细，具有一定的参考借鉴价值，需要的朋友可以参考下
2019-05-05
Docker容器化部署Python应用过程解析
这篇文章主要介绍了Docker容器化部署Python应用过程解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2020-11-11
SpringBoot打包docker镜像发布的详细步骤
这篇文章主要介绍了SpringBoot打包docker镜像发布,本文通过图文并茂的形式给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
2022-04-04
docker引起服务器磁盘爆满的问题解决
本文主要介绍了docker引起服务器磁盘爆满的问题解决，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2022-06-06
docker-desktop启动k8s的详细方法
这篇文章主要介绍了docker-desktop启动k8s,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
2022-08-08
Docker 部署 MongoDB容器的方法
Docker就可以帮助我们将开发的应用程序，以及该程序的依赖包都打包在一个文件里，下面通过本文给大家分享Docker 部署 MongoDB的详细教程，感兴趣的朋友一起看看吧
2021-06-06
Windows下Docker安装各种软件的详细过程
这篇文章主要介绍了Windows下Docker安装各种软件的详细过程，本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
2021-07-07