使用nvidia-docker创建gpu支持深度学习环境的容器

初次接触nvidia-docker相关的内容,记录部署的步骤和尝试过程。

一、安装nvidia-docker并测试

1、首先安装docker和nvidia-doker;
2、vim /etc/docker/daemon.json,添加代红框中的字段,这是为了在执行docker命令时默认使用nvidia-docker;修改好之后启动docker和nvidia-docker服务。
在这里插入图片描述

3、nvcc -V查看cuda版本,使用docker pull拉取对应cuda版本的镜像;
在这里插入图片描述
我这里是:docker pull nvidia/cuda:10.0-base
在这里插入图片描述
4、创建一个基于cuda镜像的容器,测试nvidia-smi
docker run --runtime=nvidia --rm n

  • 1
    点赞
  • 20
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
nvidia-docker是一个可以在docker使用GPU的工具。它通过在docker上做了一层封装,并调用nvidia-docker-plugin来实现在docker使用GPU的功能。在安装nvidia-docker之前,需要先安装docker。安装nvidia-docker的步骤如下: 1. 安装docker使用以下命令安装docker: ``` sudo apt-get install docker ``` 2. 安装NVIDIA驱动:访问NVIDIA驱动官网,下载并安装适合你的显卡的驱动。安装完成后,可以使用以下命令检查是否安装成功: ``` nvidia-smi ``` 3. 安装nvidia-docker使用以下命令安装nvidia-docker: ``` distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \ && curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - \ && 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 -y nvidia-docker2 sudo systemctl restart docker ``` 4. 重启docker使用以下命令重启docker: ``` sudo systemctl restart docker ``` 安装完成后,你就可以使用nvidia-docker来在docker使用GPU了。 #### 引用[.reference_title] - *1* *3* [安装使用NVIDIA-Docker——可使用GPUDocker容器](https://blog.csdn.net/m0_67390379/article/details/124477868)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control_2,239^v3^insert_chatgpt"}} ] [.reference_item] - *2* [深度学习docker环境配置之nvidia-docker安装使用](https://blog.csdn.net/qq_41776453/article/details/129794608)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control_2,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

蔡逸超

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值