Commit 93a2ed28 authored by Samuel GAIST's avatar Samuel GAIST

[dock] Only enable GPU if available

Some execution environments may run CPU as well as GPU
code, this check for Nvidia driver presence will allow
to reuse the same environment on both.

It also updates the GPU option to use the latest version
provided by the nvidia-docker2 documentation.
parent 586b35f3
Pipeline #37865 passed with stage
in 42 minutes and 33 seconds
......@@ -506,7 +506,8 @@ class Host(object):
if ("capabilities" in image_infos) and (
"gpu" in image_infos["capabilities"]
):
cmd.append("--runtime=nvidia")
if os.path.exists("/proc/driver/nvidia"):
cmd.append("--gpus=all")
if virtual_memory_in_megabytes:
# For this to work properly, memory swap limitation has to be
......
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