Commit 2b1e3375 authored by Daniele Venzano's avatar Daniele Venzano

Workaround bug in nvidia-docker: https://github.com/NVIDIA/nvidia-docker/issues/515

parent 13ec2f68
......@@ -208,7 +208,8 @@ def preload_image(image_name):
def update_service_resource_limits(service, cores=None, memory=None):
"""Update a service reservation."""
backend = _get_backend()
backend.update_service(service, cores, memory)
if 'gpu' not in service.labels: # see https://github.com/NVIDIA/nvidia-docker/issues/515
backend.update_service(service, cores, memory)
def node_list():
......
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