在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...
  • qq_35254726
  • qq_35254726
  • 2017年02月25日 16:11
  • 3238

Docker 安装 TensorFlow GPU 实战

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

在Docker下使用Nvidia GPU进行计算

【系统:CentOS 7, Nvidia Titan X】 最近在使用GPU计算,同时也在使用docker做虚拟化环境。那么问题来了,在虚拟机下如何使用GPU呢?在网上找到了很多答案,例如使用LXC...
  • bychahaha
  • bychahaha
  • 2015年09月16日 14:07
  • 6278

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

nvidia-docker快速迁移caffe环境(GPU+VNCserver+lxde桌面)
  • u013010889
  • u013010889
  • 2017年02月22日 23:05
  • 4717

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

主要包括两部分: 1. Docker与NVIDIA-Docker的安装与配置 2. Caffe镜像使用 1 Docker与NVIDIA-Docker的安装与配置 1.1 Docker安...
  • oJiMoDeYe12345
  • oJiMoDeYe12345
  • 2017年05月20日 15:30
  • 8378

k8s调用gpu,nvidia-docker使用gpu

简单介绍了nvidia-docker,介绍了k8s如何调用gpu
  • qq_35254726
  • qq_35254726
  • 2017年03月04日 15:00
  • 3692

Docker - 基于Docker的深度学习

基于Docker的深度学习原文Docker提供了一种静态链接Linux核到应用程序的方式. 采用Docker容器可以调用GPUs,因此对于Tensorflow或者其它机器学习框架的部署是一种很好的工具...
  • oJiMoDeYe12345
  • oJiMoDeYe12345
  • 2017年05月24日 15:20
  • 2420

别为Docker本地实现不支持GPU发愁,解决方案在此!

导读      通过提供独立的执行环境而不需要整个虚拟机的开销,容器已经成为大规模部署应用程序的很有吸引力的选择。 Docker让容器变得易于使用,因此受到欢迎。通过使多个工...
  • lQ1NS259ej3OKYvK4Jf
  • lQ1NS259ej3OKYvK4Jf
  • 2017年12月08日 00:00
  • 444

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

Caffe的GPU环境搭建比较复杂,所需依赖如下,本文全部会说明安装方法 GNU/Linux x86_64 with kernel version > 3.10Docker >= 1.9 (off...
  • u011987514
  • u011987514
  • 2017年04月29日 02:16
  • 2464

Ubuntu 16.04安装CUDA9+Docker CE+NVIDIA-Docker+TensorFlow/XGBoost

Ubuntu 16.04安装CUDA9+Docker CE+NVIDIA-Docker+TensorFlow/XGBoost Leave a reply 以Ubuntu16.04系统为例(...
  • baobei0112
  • baobei0112
  • 2018年01月11日 10:52
  • 230
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:在Docker下使用Nvidia GPU进行计算
举报原因:
原因补充:

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