在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