Merge branch 'rc1.7.0' of github.com:codalab/codalab-worksheets into …

…rc1.7.0
codalab · Aug 24, 2023 · 23c2d3a · 23c2d3a
2 parents 74a4fc4 + 6f26c50
commit 23c2d3a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)