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部署hadoop集群的详细教程
这篇文章主要介绍了使用docker部署hadoop集群的详细教程,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值需要的朋友可以参考下
2020-09-09
docker下安装Nginx的方法
这篇文章主要介绍了docker下安装Nginx的方法，非常不错，具有一定的参考借鉴价值,需要的朋友可以参考下
2018-07-07
从进程中去理解 Docker隔离技术
这篇文章主要介绍了从进程中去理解 Docker隔离技术，Docker 公司发布的 Docker 项目具有里程碑式的意义；Docker 项目通过“容器镜像”，解决了应用打包这个根本性难题，下文更多相关资料,需要的朋友可以参考一下
2022-04-04
使用Docker快速搭建你的Gitbook
这篇文章主要介绍了使用Docker快速搭建你的Gitbook的相关资料,需要的朋友可以参考下
2023-11-11
Docker创建openresty容器的实现步骤
OpenResty可以用来构建高性能的Web应用和服务,本文主要介绍了Docker创建openresty容器的实现步骤,文中通过示例代码介绍的非常详细,需要的朋友们下面随着小编来一起学习学习吧
2024-03-03
docker容器里面的root权限获取方法
这篇文章主要介绍了docker容器里面的root权限获取方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2021-03-03
docker修改容器配置文件的3种方法总结
有时候可能需要修改运行在docker容器中的nginx的配置文件,或者其他一些已经运行和启动很久的容器中的配置文件,下面这篇文章主要给大家介绍了关于docker修改容器配置文件的3种方法,需要的朋友可以参考下
2022-04-04
docker容器下配置jupyter notebook的操作
这篇文章主要介绍了docker容器下配置jupyter notebook的操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2021-03-03
如何将本地构建好的docker镜像发布到dockerhub
这篇文章主要介绍了如何将本地构建好的docker镜像发布到dockerhub,文中通过图文教程介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2020-05-05
一文带你快速了解和使用Docker
本文对 Docker 进行全面阐述，详细介绍 Docker 的作用、其基本使用，如常用命令、Dockerfile 的作用及使用、Docker Compose 的作用及使用，感兴趣的同学跟着小编一起来学习吧
2023-05-05