diff --git a/atlas_server/src/atlas-server/__main__.py b/atlas_server/src/atlas-server/__main__.py index e95d0b8b1..a57dc513d 100644 --- a/atlas_server/src/atlas-server/__main__.py +++ b/atlas_server/src/atlas-server/__main__.py @@ -9,6 +9,7 @@ from threading import Thread from time import sleep from urllib.parse import urlparse +from pkg_resources import parse_version import docker import yaml @@ -416,7 +417,7 @@ def _container_specs(self, args, num_workers, cuda_devices): 'mem_limit': '300m' } - if args.enable_gpu: + if args.enable_gpu and parse_version(self._client.version()['Version']) < parse_version('19.03'): scheduler_spec['runtime'] = 'nvidia' specs.append(scheduler_spec)