From 6f26c509c76c5be0109d2f81f30f0999ad95b28b Mon Sep 17 00:00:00 2001 From: Ashwin Ramaswami Date: Wed, 23 Aug 2023 18:18:17 -0400 Subject: [PATCH] Revert to sulfurheron/nvidia-cuda:9.0-cudnn7-devel-ubuntu16.04-2018-06-08 --- codalab/worker/docker_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codalab/worker/docker_utils.py b/codalab/worker/docker_utils.py index c547ab94a..aab43f76c 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)