Commit 64a773df authored by Samuel GAIST's avatar Samuel GAIST
Browse files

[dock] Use --runtime for nvidia enabled container

The nvidia-docker2 package recommends to use the --runtime
parameter to select the nvidia runtime when using a GPU
enabled container with CUDA.
parent 057de86e
......@@ -337,7 +337,7 @@ class Host(object):
if container.image in Host.images_cache:
image_infos = Host.images_cache[container.image]
if ('capabilities' in image_infos) and ('gpu' in image_infos['capabilities']):
cmd[0] = 'nvidia-docker'
cmd.append('--runtime=nvidia')
if virtual_memory_in_megabytes:
......
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