nvidia-docker에서 GPU 사용을 제한 하시겠습니까?

Question

멀티 GPU 서버에 내부 Jupyterhub를 설치하려고합니다. Jupyter 액세스는 도커 인스턴스를 통해 제공됩니다. 각 사용자의 액세스를 단일 GPU 이상으로 제한하고 싶습니다. 나는 어떤 제안이나 의견을 주시면 감사하겠습니다. 감사.

Answer 1

당신은 다음과 같이 그냥 "엔비디아 고정 표시기"전에 환경 변수 "NV_GPU"를 추가, 문제는이 방법으로 해결 될 수 nvidia-docker-compose

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

Answer 2

그것을 시도 할 수 있습니다 :

---

[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