Docker上的“nvidia-smi”不显示“进程”是否正确？答案

【问题标题】：Is it correct that "nvidia-smi" on Docker does not show "Processes"?Docker上的“nvidia-smi”不显示“进程”是否正确？
【发布时间】：2020-08-30 07:09:21
【问题描述】：

当我在 Docker 上运行“nvidia-smi”时，它没有显示在“进程”部分中是否正常？

我正在 ubuntu 上使用 Docker + GPU 在 ubuntu 上构建深度学习环境。

我认为它几乎完成了，但有一件事情困扰着我。当我在 Ubuntu 上执行“nvidia-smi”时，我看到了“进程”。

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 440.100      Driver Version: 440.100      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 GTX 166...  Off  | 00000000:01:00.0  On |                  N/A |
| 42%   37C    P8     8W / 125W |    249MiB /  5936MiB |      3%      Default |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|    0      1127      G   /usr/lib/xorg/Xorg                            35MiB |
|    0      2006      G   /usr/lib/xorg/Xorg                            94MiB |
|    0      2202      G   /usr/bin/gnome-shell                          97MiB |
|    0      6565      G   /usr/lib/firefox/firefox                       2MiB |
|    0      7875      G   /usr/lib/firefox/firefox                       2MiB |
|    0     10070      G   /usr/lib/firefox/firefox                       2MiB |
+-----------------------------------------------------------------------------+

当我在 Docker 上执行“nvidia-smi”时，我看不到“进程”。

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 440.100      Driver Version: 440.100      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 GTX 166...  Off  | 00000000:01:00.0  On |                  N/A |
| 42%   36C    P8     8W / 125W |    342MiB /  5936MiB |      2%      Default |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
+-----------------------------------------------------------------------------+

当我运行“Jupyter Notebook”时，GPU 似乎正在运行。

“这是一个规范，”我读到几年前写的一篇文章。

是不是 Process 今天仍未出现的“规范”？还是因为我没有正确安装？

借给我你的智慧。提前感谢！

【问题讨论】：

标签： docker ubuntu nvidia

【解决方案1】：

是的，您将无法看到，因为驱动程序不知道 PID 命名空间。您可以仔细阅读线程和使用 Python 的解决方法，特别是在

https://github.com/NVIDIA/nvidia-docker/issues/179#issuecomment-598059213

（我假设您没有使用 VM，因为显示的日志中的持久性模式已关闭）。

【讨论】：

【解决方案2】：

一个 shim 驱动程序允许 in-docker nvidia-smi 显示正确的进程列表而无需修改任何内容。

https://github.com/matpool/mpu

【讨论】：