From ca1c6831d9c5b83c374720dc8d82c3f6ffaa204b Mon Sep 17 00:00:00 2001 From: Yonas Tesfaye Date: Thu, 2 Feb 2017 18:12:37 -0800 Subject: [PATCH] Make sure that gpu always maps to GPU 0 inside the contianer --- l41_nbhub/MarathonSpawner.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/l41_nbhub/MarathonSpawner.py b/l41_nbhub/MarathonSpawner.py index 67d9d59..ff9339a 100644 --- a/l41_nbhub/MarathonSpawner.py +++ b/l41_nbhub/MarathonSpawner.py @@ -130,7 +130,7 @@ def start(self): parameters = [{'key':'workdir', 'value':os.path.join(self.home_basepath, self.user.name)}] parameters.append({'key': 'device', 'value': '/dev/nvidiactl'}) parameters.append({'key': 'device', 'value': '/dev/nvidia-uvm'}) - parameters.append({'key': 'device', 'value': '/dev/nvidia%d'%gpu_id}) + parameters.append({'key': 'device', 'value': '/dev/nvidia%d:/dev/nvidia0'%gpu_id}) parameters.append({'key': 'volume-driver', 'value': 'nvidia-docker'}) parameters.append({'key': 'volume', 'value': 'nvidia_driver_{}:/usr/local/nvidia:ro'.format(driver_version)}) cmd = "/bin/bash /srv/ganymede_nbserver/ganymede_nbserver.sh"