docker支持gpu么

docker支持gpu，docker可以通过nvidia-docker2来使用GPU。在daemon.json文件中配置runtime使用nvidia，启动容器后运行nvidia-smi就能看到所有的GPU。

docker挂载 GPU的方法介绍：

使用 nvidia-docker2

简言之，使用 nvidia-docker2，可以不费吹灰之力就能使用到 GPU，仅仅需要配置 runtime 使用 nvidia

cat /etc/docker/daemon.json {     "default-runtime": "nvidia",     "runtimes": {         "nvidia": {             "path": "/usr/bin/nvidia-container-runtime",             "runtimeArgs": []         }     },     "exec-opts": ["native.cgroupdriver=systemd"] }

启动容器之后，运行 nvidia-smi 能看到所有的 GPU 卡：

[root@localhost] docker run -it 98b41a1e975d bash root@6db1dd28459d:/notebooks# nvidia-smi  +-----------------------------------------------------------------------------+ | NVIDIA-SMI 410.79       Driver Version: 410.79       CUDA Version: 10.0     | |-------------------------------+----------------------+----------------------+ | GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC | | Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. | |===============================+======================+======================| |   0  Tesla V100-SXM2...  On   | 00000000:8A:00.0 Off |                    0 | | N/A   40C    P0    57W / 300W |   4053MiB / 16130MiB |      4%      Default | +-------------------------------+----------------------+----------------------+ |   1  Tesla V100-SXM2...  On   | 00000000:8B:00.0 Off |                    0 | | N/A   38C    P0    40W / 300W |      0MiB / 16130MiB |      0%      Default | +-------------------------------+----------------------+----------------------+ |   2  Tesla V100-SXM2...  On   | 00000000:8C:00.0 Off |                    0 | | N/A   42C    P0    46W / 300W |      0MiB / 16130MiB |      0%      Default | +-------------------------------+----------------------+----------------------+ |   3  Tesla V100-SXM2...  On   | 00000000:8D:00.0 Off |                    0 | | N/A   39C    P0    40W / 300W |      0MiB / 16130MiB |      0%      Default | +-------------------------------+----------------------+----------------------+ |   4  Tesla V100-SXM2...  On   | 00000000:B3:00.0 Off |                    0 | | N/A   39C    P0    42W / 300W |      0MiB / 16130MiB |      0%      Default | +-------------------------------+----------------------+----------------------+ |   5  Tesla V100-SXM2...  On   | 00000000:B4:00.0 Off |                    0 | | N/A   41C    P0    57W / 300W |   7279MiB / 16130MiB |      4%      Default | +-------------------------------+----------------------+----------------------+ |   6  Tesla V100-SXM2...  On   | 00000000:B5:00.0 Off |                    0 | | N/A   40C    P0    45W / 300W |      0MiB / 16130MiB |      0%      Default | +-------------------------------+----------------------+----------------------+ |   7  Tesla V100-SXM2...  On   | 00000000:B6:00.0 Off |                    0 | | N/A   41C    P0    44W / 300W |      0MiB / 16130MiB |      0%      Default | +-------------------------------+----------------------+----------------------+  +-----------------------------------------------------------------------------+ | Processes:                                                       GPU Memory | |  GPU       PID   Type   Process name                             Usage      | |=============================================================================| +-----------------------------------------------------------------------------+

通过 NVIDIA_DRIVER_CAPABILITIES 可以加入部分的 library。通过 NVIDIA_VISIBLE_DEVICES 可以只使用某些 GPU 卡

[root@localhost cuda-9.0]# docker run -it  --env NVIDIA_DRIVER_CAPABILITIES="compute,utility"  --env NVIDIA_VISIBLE_DEVICES=0,1 98b41a1e975d bash root@97bf127ff83a:/notebooks# nvidia-smi Tue Oct 15 09:29:45 2019 +-----------------------------------------------------------------------------+ | NVIDIA-SMI 410.79       Driver Version: 410.79       CUDA Version: 10.0     | |-------------------------------+----------------------+----------------------+ | GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC | | Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. | |===============================+======================+======================| |   0  Tesla V100-SXM2...  On   | 00000000:8A:00.0 Off |                    0 | | N/A   39C    P0    57W / 300W |   4053MiB / 16130MiB |      3%      Default | +-------------------------------+----------------------+----------------------+ |   1  Tesla V100-SXM2...  On   | 00000000:8B:00.0 Off |                    0 | | N/A   37C    P0    40W / 300W |      0MiB / 16130MiB |      0%      Default | +-------------------------------+----------------------+----------------------+  +-----------------------------------------------------------------------------+ | Processes:                                                       GPU Memory | |  GPU       PID   Type   Process name                             Usage      | |=============================================================================| +-----------------------------------------------------------------------------+

更多相关教程，请关注PHP中文网docker教程栏目。

以上就是docker支持gpu么的详细内容，更多内容请关注技术你好其它相关文章！