Ubuntu18.04 nvidia-dockerのインストール(親測定有効、手順詳細)

3257 ワード

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/996706421.次の手順に従います.
$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-docker 2パッケージをインストールし、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

成功しました.