docker容器使用GPU方法实现

 更新时间:2022年05月25日 14:38:18   作者:m0_54850467  
本文主要介绍了docker容器使用GPU方法实现,文中通过示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下

为了使docker容器可以使用nvidia-docker,可以有如下几种办法:

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内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

相关文章

  • CentOS7中配置Docker的yum源并安装使用详解

    CentOS7中配置Docker的yum源并安装使用详解

    这篇文章主要介绍了CentOS7中配置Docker的yum源并安装使用详解,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2019-09-09
  • Docker容器无法解析域名的处理

    Docker容器无法解析域名的处理

    这篇文章主要介绍了Docker容器无法解析域名的处理问题,具有很好的参考价值,希望对大家有所帮助,
    2023-11-11
  • Docker 存储管理的几种方式

    Docker 存储管理的几种方式

    本文主要使用三种不同的方式将数据从Docker主机挂载到容器中,分别为卷(volumes),绑定挂载(bind mounts),临时文件系tmpfs,还介绍了数据卷容器、数据卷的备份与恢复,对Docker 存储管理相关知识感兴趣的朋友一起看看吧
    2022-07-07
  • docker-maven-plugin 插件无法拉取对应jar包问题

    docker-maven-plugin 插件无法拉取对应jar包问题

    这篇文章主要介绍了docker-maven-plugin 插件无法拉取问题,总是报错,如何解决这个问题呢,下面小编给大家带来了解决方法,一起看看吧
    2021-09-09
  • Docker 容器操作指令汇总详解

    Docker 容器操作指令汇总详解

    这篇文章主要介绍了Docker 容器操作指令汇总详解,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2019-10-10
  • 使用docker快速部署Nginx、Redis、MySQL、Tomcat及制作镜像的方法

    使用docker快速部署Nginx、Redis、MySQL、Tomcat及制作镜像的方法

    这篇文章介绍了如何使用Docker快速部署Nginx、Redis、MySQL、Tomcat以及制作镜像的过程,它涵盖了从基础镜像的使用、Dockerfile的编写到应用的构建和部署的详细步骤,通过本文,可以掌握如何利用Docker简化应用的部署和管理,感兴趣的朋友跟随小编一起看看吧
    2024-11-11
  • Docker端口占用问题分析以及解决方案

    Docker端口占用问题分析以及解决方案

    这篇文章主要介绍了Docker端口占用问题分析以及解决方案,主要步骤包括检查端口占用、处理Windows环境下的特殊问题、解决权限问题、修改端口映射和考虑其他可能原因,文中通过代码介绍的非常详细,需要的朋友可以参考下
    2025-04-04
  • 使用docker启动redis守护进程的详细流程

    使用docker启动redis守护进程的详细流程

    在现代软件开发中,缓存和消息队列是提高应用性能和扩展性的关键,Redis 是一种流行的内存数据结构存储系统,使用 Docker 容器化技术可以方便地部署和管理 Redis 服务,本文给大家介绍了使用docker启动redis守护进程的详细流程,需要的朋友可以参考下
    2025-01-01
  • IDEA直接打包Docker镜像的实现

    IDEA直接打包Docker镜像的实现

    Docker提供了多种方法来打包镜像,本文主要介绍了IDEA直接打包Docker镜像的实现,具有一定的参考价值,感兴趣的可以了解一下
    2024-06-06
  • 使用OpenSSL生成Kubernetes证书的介绍

    使用OpenSSL生成Kubernetes证书的介绍

    今天小编就为大家分享一篇关于使用OpenSSL生成Kubernetes证书的介绍,小编觉得内容挺不错的,现在分享给大家,具有很好的参考价值,需要的朋友一起跟随小编来看看吧
    2019-01-01

最新评论