Ubuntu18.04安装nvidia-docker(亲测有效,步骤详尽)

42 篇文章 2 订阅

Ubuntu 18.04安装NVIDIA-docker

0.前提: 
  (1)首先安装docker;具体步骤:https://blog.csdn.net/BigData_Mining/article/details/87868711
  (2)安装Nvidia驱动,CUDA和CUDNN;具体步骤:https://blog.csdn.net/BigData_Mining/article/details/99670642
1.然后执行下面的步骤:

$curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
$curl -s -L https://nvidia.github.io/nvidia-docker/ubuntu18.04/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
$sudo apt-get update

2.安装nvidia-docker2软件包并重新加载docker守护程序配置

$sudo apt-get install nvidia-docker2
$sudo pkill -SIGHUP dockerd
$docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi

[注]:运行上一步若报错:

qyh@qyh-mas:~$ sudo docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi
docker: Error response from daemon: Unknown runtime specified nvidia.
See 'docker run --help'.

当显卡驱动安装好的前提下(用指令$nvidia-smi来显示显卡驱动),则换成运行下步,指定安装的cuda版本:

$sudo nvidia-docker run --rm nvidia/cuda:10.1-devel nvidia-smi
qyh@qyh-mas:~$ sudo nvidia-docker run --rm nvidia/cuda:10.1-devel | nvidia-smi
Fri Aug 16 16:30:02 2019       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 430.26       Driver Version: 430.26       CUDA Version: 10.2     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  GeForce 940MX       Off  | 00000000:3C:00.0 Off |                  N/A |
| N/A   52C    P0    N/A /  N/A |    239MiB /  2004MiB |      4%      Default |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|    0      3038      G   /usr/lib/xorg/Xorg                           121MiB |
|    0      4709      G   /usr/bin/gnome-shell                         109MiB |
|    0     18218      G   /usr/lib/firefox/firefox                       1MiB |
|    0     18402      G   /usr/lib/firefox/firefox                       1MiB |
|    0     18463      G   /usr/lib/firefox/firefox                       1MiB |
+-----------------------------------------------------------------------------+
Unable to find image 'nvidia/cuda:10.1-devel' locally
10.1-devel: Pulling from nvidia/cuda
7413c47ba209: Pulling fs layer
0fe7e7cbb2e8: Pulling fs layer
1d425c982345: Pulling fs layer
344da5c95cec: Pulling fs layer
43bcc41986db: Pulling fs layer
76661327d908: Pulling fs layer
abdc887b90e5: Pulling fs layer
eb38470de0e2: Pulling fs layer
0adb8d7f107f: Pulling fs layer

表示成功了!

  • 9
    点赞
  • 104
    收藏
    觉得还不错? 一键收藏
  • 19
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 19
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值