在Docker下使用Nvidia GPU进行计算

【系统: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
阅读更多
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭
关闭