docker支持gpu么

Docker  /  管理员 发布于 7年前   232

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 bashroot@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 bashroot@97bf127ff83a:/notebooks# nvidia-smiTue 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      ||=============================================================================|+-----------------------------------------------------------------------------+

更多相关教程，请关注docker教程栏目。

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