Revert to sulfurheron/nvidia-cuda:9.0-cudnn7-devel-ubuntu16.04-2018-0…

…6-08
codalab · Aug 23, 2023 · 6f26c50 · 6f26c50
1 parent f4ff22e
commit 6f26c50
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)