在Docker下使用Nvidia GPU进行计算

原创 2015年09月16日 14:07:40

【系统:CentOS 7, Nvidia Titan X】
最近在使用GPU计算,同时也在使用docker做虚拟化环境。那么问题来了,在虚拟机下如何使用GPU呢?

在网上找到了很多答案,例如使用LXC等。

实际上将GPU映射到Docker下很容易,只需要在启动镜像的时候将设备挂载上去即可。

  1. 查看显卡设备名
[root@XXX ~]# ls -la /dev | grep nvidia
crw-rw-rw-.  1 root root    195,   0 Sep 16 13:49 nvidia0
crw-rw-rw-.  1 root root    195, 255 Sep 16 13:49 nvidiactl
crw-rw-rw-.  1 root root    247,   0 Sep 16 13:54 nvidia-uvm
  1. 挂载对应设备,启动虚拟机
docker run -it --name NAME -v /home/:/mnt/home --privileged=true --device /dev/nvidia-uvm:/dev/nvidia-uvm --device /dev/nvidia0:/dev/nvidia0 --device /dev/nvidiactl:/dev/nvidiactl myconda:cuda bash

这里注意需要使用 privilege 模式,每个设备都用一次 --device 挂载

  1. 虚拟机内安装驱动以及CUDA

相关文章推荐

docker简单操作,以及运行gpu

最近正在做关于gpu的运行相关的工作。所以本篇文章所写的都是本人经过真实操作的,并非网络版本的copy,这个也是增加积累以及分享。          先说下我最近的目的,希望训练模型可以运行在cpu...

Docker 安装 TensorFlow GPU 实战

安装背景AI如雨后春笋般的出现,DEVOPS的理论不断深入。所有高大上的开源产品都支持两个环境:docker 和Linux。本文主要讲解怎么在一台安装了GPU的centos7 环境安装tensorfl...

k8s调用gpu,nvidia-docker使用gpu

简单介绍了nvidia-docker,介绍了k8s如何调用gpu

Docker - 基于NVIDIA-Docker的Caffe-GPU环境搭建

Docker 基于NVIDIA-Docker的Caffe-GPU环境搭建

Ubuntu16.04下基于Docker的Caffe-GPU版本环境搭建总结

Caffe的GPU环境搭建比较复杂,所需依赖如下,本文全部会说明安装方法 GNU/Linux x86_64 with kernel version > 3.10Docker >= 1.9 (off...

Docker run 命令的使用方法

Docker run 命令的使用方法 注意,本文基于最新的Docker 1.4文档翻译。 Docker会在隔离的容器中运行进程。当运行 docker run命令时,Docker会启...
  • wongson
  • wongson
  • 2015年10月12日 17:44
  • 30520

nvidia-docker快速迁移caffe环境(GPU+VNCserver+lxde桌面)

nvidia-docker快速迁移caffe环境(GPU+VNCserver+lxde桌面)

Docker on Windows初体验

Docker on Windows初体验写在前面:之前想在WIN 10 系统装TensoFlow,所以follow到了Docker这个黑科技,花了一天时间安装熟悉了一下。如果需要在多台机器需要部署环境...

在Windows上玩TensorFlow(一)——安装Docker

“谷歌”+“深度学习”,两个标签让2015年12月才由谷歌开源的深度学习工具TensorFlow在其发布之后就迅速地成为了全球最为炙手可热的开源项目,2016年4月,开源的TensorFlow又支持了...

【Docker】在ubuntu14.04镜像上安装GPU显卡驱动, CUDA和CUDNN

1. pull ubuntu 14.04镜像 sudo docker pull ubuntu:14.04 2. 安装nvidia-docker 地址:https://github.com/NVIDIA...
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:在Docker下使用Nvidia GPU进行计算
举报原因:
原因补充:

(最多只允许输入30个字)