关闭

在Docker下使用Nvidia GPU进行计算

4455人阅读 评论(0) 收藏 举报

【系统: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
0
1

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:12215次
    • 积分:189
    • 等级:
    • 排名:千里之外
    • 原创:6篇
    • 转载:0篇
    • 译文:0篇
    • 评论:3条
    最新评论