Commit 386fe36f authored by Philip ABBET's avatar Philip ABBET
Browse files

Add support for 'nvidia-docker'

parent b07c7620
......@@ -330,6 +330,12 @@ 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'
if virtual_memory_in_megabytes:
# For this to work properly, memory swap limitation has to be enabled on
# the kernel. This typically goes by setting "cgroup_enable=memory" as a
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