nvidia-docker的一次实际应用

本文介绍了如何使用nvidia-docker在GPU环境下运行深度学习代码。内容涉及nvidia-docker的安装过程,包括安装docker、nvidia显卡驱动和nvidia-docker,以及启动和配置nvidia-docker的步骤。强调了在启动docker容器时需使用nvidia-docker以利用GPU资源,并提示了如何修改docker的默认存储路径。
摘要由CSDN通过智能技术生成

在深度学习中,如果要使docker中的代码能够在GPU的环境下运行,需要安装nvidia-docker。nvidia-docker是一个可以是用GPU的docker,其在docker上做了一层封装,通过nvidia-docker-plugin,然后调用到docker上,其最终实现还是在docker的启动命令上携带一些必要的参数,因此在安装nvidia-docker之前,还是需要安装docker的。

单纯的docker,一般是使用CPU的应用,如果要使用GPU的话, 就需要安装特定硬件环境,所以docker并不能支持Nvidia GPU,为了解决这个问题,最初的作法是进入docker容器,全部重新安装nvidia driver。然后通过设置相应的设备参数来启动容器,不过这种方法需要宿主机的driver版本和容器内的driver版本完全一致,导致docker镜像无法共享,所以这种方法不能大规模使用。

为了使docker image能很方便的使用Nvidia GPU,从而出现了nvidia-docker,使用nvidia-docker制作的docker 镜像,就要求在启动容器的时候其需要的字符设备和驱动程序必须已经被挂载。

 

nvidia-docker-plugin是一个docker plugin,被用来帮助我们轻松部署container到GPU混合的环境下。类似一个守护进程,发现宿主机驱动文件以及GPU 设备,并且将这些挂载到来自docker守护进程的请求中。以此来支持docker GPU的使用。

 

使用nvidia-docker需要安装的软件:

1、docker

2、nvidia显卡驱动

3、nvidia docker

3.1、下载nvidia-docker.repo文件,并将该文件输出到/etc/yum.repos.d/nvidia-docker.repo

执行:curl -s -L https://nvidia.github.io/nvidia-docker/centos7/x86_64/nvidia-docker.repo | sudo tee /etc/yum.repos.d/nvidia-docker.repo

结果如下:

 

3.2 查找可以安装的nvidia-docker

执行:y

  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
### 回答1: 要在Ubuntu 20中离线安装NVIDIA Docker,您需要进行以下步骤: 1. 首先,您需要下载NVIDIA Docker的离线安装包。您可以在NVIDIA Docker的官方GitHub页面上找到不同版本的离线安装包。 2. 在下载完离线安装包后,将其解压缩到您选择的目录中。 3. 接下来,您需要安装Docker引擎。可以通过运行以下命令来完成安装: ``` $ sudo apt-get update $ sudo apt-get install docker.io ``` 4. 安装完成后,您可以运行以下命令来验证Docker引擎是否正确安装: ``` $ sudo systemctl status docker ``` 5. 确保NVIDIA驱动程序已经安装在您的系统中。您可以通过运行以下命令来检查: ``` $ nvidia-smi ``` 6. 接下来,您需要将NVIDIA Docker的可执行文件路径添加到系统的环境变量中。可以通过编辑~/.bashrc文件并在末尾添加以下行来实现: ``` $ export PATH=/path/to/nvidia-docker:$PATH ``` 7. 保存并关闭~/.bashrc文件后,运行以下命令以使其生效: ``` $ source ~/.bashrc ``` 8. 现在,您可以运行以下命令来测试NVIDIA Docker是否成功安装并正在运行: ``` $ nvidia-docker run --rm nvidia/cuda nvidia-smi ``` 如果您得到了有关NVIDIA GPU的信息,那么恭喜您成功地在Ubuntu 20上离线安装了NVIDIA Docker。现在您可以使用NVIDIA容器来运行GPU加速的Docker应用程序。 ### 回答2: 要在Ubuntu 20离线安装NVIDIA Docker,需要执行以下步骤: 1. 首先,确保你已经下载了NVIDIA Docker的离线安装包以及相关的依赖文件,并将它们保存在一个可访问的位置。 2. 打开终端,切换到保存下载文件的目录。 3. 安装Docker Engine。运行以下命令: ``` sudo apt-get remove docker docker-engine docker.io containerd runc sudo apt-get update sudo apt-get install apt-transport-https ca-certificates curl gnupg lsb-release curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo gpg --dearmor -o /usr/share/keyrings/docker-archive-keyring.gpg echo "deb [arch=amd64 signed-by=/usr/share/keyrings/docker-archive-keyring.gpg] https://download.docker.com/linux/ubuntu $(lsb_release -cs) stable" | sudo tee /etc/apt/sources.list.d/docker.list > /dev/null sudo apt-get update sudo apt-get install docker-ce docker-ce-cli containerd.io ``` 4. 安装NVIDIA runtime。运行以下命令: ``` curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - distribution=$(. /etc/os-release;echo $ID$VERSION_ID) curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list sudo apt-get update sudo apt-get install nvidia-docker2 sudo systemctl restart docker ``` 5. 验证安装是否成功。运行以下命令: ``` sudo docker run --rm --gpus all nvidia/cuda:11.0-base nvidia-smi ``` 以上就是在Ubuntu 20中离线安装NVIDIA Docker的简要步骤。确保按照步骤逐一执行,并根据你的实际情况进行相应的调整。 ### 回答3: 要在Ubuntu 20上离线安装NVIDIA Docker,需要按照以下步骤进行操作: 1. 首先,从NVIDIA官网下载适用于Ubuntu 20的NVIDIA Docker包,将其保存到一台可联网的机器上。 2. 在这台可联网的机器上,安装依赖软件包。打开终端,执行以下命令: ``` sudo apt-get update sudo apt-get install -y apt-transport-https ca-certificates curl software-properties-common ``` 3. 添加NVIDIA Docker的GPG密钥。执行以下命令: ``` curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add - ``` 4. 添加Docker存储库。执行以下命令: ``` sudo add-apt-repository "deb [arch=amd64] https://download.docker.com/linux/ubuntu focal stable" ``` 5. 重新更新一次软件包并安装Docker。执行以下命令: ``` sudo apt-get update sudo apt-get install -y docker-ce docker-ce-cli containerd.io ``` 6. 使用Docker启用nvidia-runtime仓库。执行以下命令: ``` curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - distribution=$(. /etc/os-release;echo $ID$VERSION_ID) curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list sudo apt-get update ``` 7. 离线导入NVIDIA Docker的GPG密钥。将在第1步中下载的密钥文件拷贝到离线机器上,并使用以下命令导入密钥: ``` sudo apt-key add <path-to-key-file> ``` 8. 执行离线安装。执行以下命令: ``` sudo apt-get update sudo apt-get install -y nvidia-docker2 ``` 9. 配置Docker以使用NVIDIA Docker运行时。执行以下命令: ``` sudo systemctl restart docker ``` 至此,您已成功在Ubuntu 20上离线安装NVIDIA Docker,并可以使用Docker运行具有GPU支持的容器。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值