diff --git a/codalab/worker/docker_utils.py b/codalab/worker/docker_utils.py index 640f1a4d3..e21dfb5eb 100644 --- a/codalab/worker/docker_utils.py +++ b/codalab/worker/docker_utils.py @@ -152,7 +152,7 @@ def get_nvidia_devices(self, use_docker=True): docker.errors.ImageNotFound if the CUDA image cannot be pulled docker.errors.APIError if another server error occurs """ - cuda_image = 'nvidia/cuda:12.2.0-devel-ubuntu22.04' + cuda_image = 'sulfurheron/nvidia-cuda:9.0-cudnn7-devel-ubuntu16.04-2018-06-08' nvidia_command = 'nvidia-smi --query-gpu=index,uuid --format=csv,noheader' if use_docker: self.client.images.pull(cuda_image)