Commit 6f48116e authored by Daniele Venzano's avatar Daniele Venzano

Add the corrcet env variables to enable the nvidia drivers

parent 6005f793
Pipeline #10413 passed with stages
in 41 seconds
......@@ -77,6 +77,7 @@ def nb_service(memory_limit, core_limit, image, gpu=False):
}
if gpu:
service['labels'] = ['gpu']
service['environment'] = [['NVIDIA_VISIBLE_DEVICES','all']]
return service
......
......@@ -14,7 +14,16 @@
"readable_descr": "README-datascience.md",
"name": "Data science notebook GPU",
"description": "pydatasci-gpu.json",
"parameters": [],
"parameters": [
{
"kind": "environment",
"name": "NVIDIA_VISIBLE_DEVICES",
"readable_name": "GPU",
"description": "Which GPU to enable for this execution (e.g. all: all GPUs, 0: just GPU #0, 0,2: GPU #0 and #2)",
"type": "string",
"default": "all"
}
],
"guest_access": false
}
]
......
......@@ -37,6 +37,6 @@ RUN pip install --upgrade pip \
torchvision
# nvidia-docker
LABEL com.nvidia.volumes.needed="nvidia_driver"
LABEL com.nvidia.cuda.version="${CUDA_VERSION}"
ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
ENV NVIDIA_REQUIRE_CUDA "cuda>=9.0"
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment