在Ubuntu上运行nvidia-docker

首先

在进行机器学习环境建设时,留下备忘录。
官方GitHub: NVIDIA/nvidia-docker
想要引入这个但是我的英文能力太差了,所以参考了下面的Qiita。
Qiita: 尝试使用nvidia-docker2

構成 – 构成

    • マシン:Mouse MASTERPIECE i1720PA1-SP2

 

    OS:Ubuntu 16.04.5 LTS

安装Ubuntu

参考文章:
Qiita:【陷阱满满!】将Ubuntu 16.04 LTS安装在MSI制造的笔记本电脑(GE62 6QC)上。

安装Docker

参考文章
Qiita: 在Ubuntu上安装docker
※离题一下,下面也一并执行
以无需sudo权限执行Docker命令的方法

安装NVIDIA驱动程序

参考文章
Qiita: 在Ubuntu上安装NVIDIA驱动的步骤

驱动程序操作确认

nvidia-smi コマンドで GeForce GTX が搭載枚数分見えていればOK

$ nvidia-smi
Fri Aug 17 15:02:51 2018
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 390.48                 Driver Version: 390.48                    |
|-------------------------------+----------------------+----------------------+
| 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 GTX 108...  Off  | 00000000:17:00.0 Off |                  N/A |
| 25%   34C    P8    12W / 250W |      2MiB / 11178MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+
|   1  GeForce GTX 108...  Off  | 00000000:65:00.0  On |                  N/A |
| 25%   41C    P8    13W / 250W |    261MiB / 11175MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|    1      1283      G   /usr/lib/xorg/Xorg                            18MiB |
|    1      1357      G   /usr/bin/gnome-shell                          49MiB |
|    1      2020      G   /usr/lib/xorg/Xorg                           104MiB |
|    1      2185      G   /usr/bin/gnome-shell                          85MiB |
+-----------------------------------------------------------------------------+

根据Qiita的指示,我已经完成对内核版本的固定设置。

安装nvidia-docker2

参考文章
在Qiita试用nvidia-docker2

基本上,可以按照上述文章的方式進行操作,只需注意自己混淆的部分。
在執行sudo apt-get install時,需要指定兩個版本名(如下)
sudo apt-get install -y nvidia-docker2={version1} nvidia-container-runtime={version2}
這取決於docker –version的結果,從sudo apt-cache madison nvidia-docker2 nvidia-container-runtime的結果中選擇。

$ sudo docker --version
Docker version 18.06.0-ce, build 0ffa825

因为我的情况下Docker的版本是18.06.0。

$ sudo apt-cache madison nvidia-docker2 nvidia-container-runtime
nvidia-docker2 | 2.0.3+docker18.06.1-1 | https://nvidia.github.io/nvidia-docker/ubuntu16.04/amd64  Packages
nvidia-docker2 | 2.0.3+docker18.06.0-1 | https://nvidia.github.io/nvidia-docker/ubuntu16.04/amd64  Packages
nvidia-docker2 | 2.0.3+docker18.03.1-1 | https://nvidia.github.io/nvidia-docker/ubuntu16.04/amd64  Packages
(中略)
nvidia-container-runtime | 2.0.0+docker18.06.1-1 | https://nvidia.github.io/nvidia-container-runtime/ubuntu16.04/amd64  Packages
nvidia-container-runtime | 2.0.0+docker18.06.0-1 | https://nvidia.github.io/nvidia-container-runtime/ubuntu16.04/amd64  Packages
nvidia-container-runtime | 2.0.0+docker18.03.1-1 | https://nvidia.github.io/nvidia-container-runtime/ubuntu16.04/amd64  Packages
(以下略)

nvidia-docker2的版本(版本名称1)是2.0.3+docker18.06.0-1,nvidia-container-runtime的版本(版本名称2)是2.0.0+docker18.06.0-1。具体命令如下所示:sudo apt-get install -y nvidia-docker2=2.0.3+docker18.06.0-1 nvidia-container-runtime=2.0.0+docker18.06.0-1。

最后

如果有任何评论或修改请求,请随时提出!

广告
将在 10 秒后关闭
bannerAds