nvidia-docker中GPU的使用限制？

Question

我在多GPU服務器上設置了一個內部Jupyterhub。 Jupyter訪問通過docker實例提供。我想限制每個用戶的訪問不超過一個GPU。我會很感激任何建議或評論。謝謝。

Answer 1

您可以nvidia-docker-compose

version: "2" 
services 
    process1: 
    image: nvidia/cuda 
    devices: 
     - /dev/nvidia0 

Answer 2

的問題可以通過這種方式來解決嘗試，只是「NVIDIA-泊塢窗」前添加環境變量「NV_GPU」如下：

---

[root@bogon ~]# NV_GPU='4,5' nvidia-docker run -dit --name tf_07 tensorflow/tensorflow:latest-gpu /bin/bash 
e04645c2d7ea658089435d64e72603f69859a3e7b6af64af005fb852473d6b56 
[root@bogon ~]# docker attach tf_07 
root@e04645c2d7ea:/notebooks# 
root@e04645c2d7ea:/notebooks# ll /dev 
total 4 
drwxr-xr-x 5 root root  460 Dec 29 03:52 ./ 
drwxr-xr-x 22 root root  4096 Dec 29 03:52 ../ 
crw--w---- 1 root tty 136, 0 Dec 29 03:53 console 
lrwxrwxrwx 1 root root  11 Dec 29 03:52 core -> /proc/kcore 
lrwxrwxrwx 1 root root  13 Dec 29 03:52 fd -> /proc/self/fd/ 
crw-rw-rw- 1 root root 1, 7 Dec 29 03:52 full 
drwxrwxrwt 2 root root  40 Dec 29 03:52 mqueue/ 
crw-rw-rw- 1 root root 1, 3 Dec 29 03:52 null 
crw-rw-rw- 1 root root 245, 0 Dec 29 03:52 nvidia-uvm 
crw-rw-rw- 1 root root 245, 1 Dec 29 03:52 nvidia-uvm-tools 
crw-rw-rw- 1 root root 195, 4 Dec 29 03:52 nvidia4 
crw-rw-rw- 1 root root 195, 5 Dec 29 03:52 nvidia5 
crw-rw-rw- 1 root root 195, 255 Dec 29 03:52 nvidiactl 
lrwxrwxrwx 1 root root  8 Dec 29 03:52 ptmx -> pts/ptmx 
drwxr-xr-x 2 root root  0 Dec 29 03:52 pts/ 
crw-rw-rw- 1 root root 1, 8 Dec 29 03:52 random 
drwxrwxrwt 2 root root  40 Dec 29 03:52 shm/ 
lrwxrwxrwx 1 root root  15 Dec 29 03:52 stderr -> /proc/self/fd/2 
lrwxrwxrwx 1 root root  15 Dec 29 03:52 stdin -> /proc/self/fd/0 
lrwxrwxrwx 1 root root  15 Dec 29 03:52 stdout -> /proc/self/fd/1 
crw-rw-rw- 1 root root 5, 0 Dec 29 03:52 tty 
crw-rw-rw- 1 root root 1, 9 Dec 29 03:52 urandom 
crw-rw-rw- 1 root root 1, 5 Dec 29 03:52 zero 
root@e04645c2d7ea:/notebooks# 

或者閱讀nvidia-docker of github's wiki