在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。
最后
如果有任何评论或修改请求,请随时提出!