Commit 2c3bc396 authored by Flavio TARSETTI's avatar Flavio TARSETTI

Merge branch 'only_enable_gpu_runtime_if_available' into 'master'

Only enable GPU runtime if available

See merge request !115
parents 586b35f3 93a2ed28
Pipeline #37873 passed with stages
in 20 minutes and 25 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