Skip to content

Commit 23c2d3a

Browse files
committed
Merge branch 'rc1.7.0' of github.com:codalab/codalab-worksheets into rc1.7.0
2 parents 74a4fc4 + 6f26c50 commit 23c2d3a

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

codalab/worker/docker_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -152,7 +152,7 @@ def get_nvidia_devices(self, use_docker=True):
152152
docker.errors.ImageNotFound if the CUDA image cannot be pulled
153153
docker.errors.APIError if another server error occurs
154154
"""
155-
cuda_image = 'nvidia/cuda:12.2.0-devel-ubuntu22.04'
155+
cuda_image = 'sulfurheron/nvidia-cuda:9.0-cudnn7-devel-ubuntu16.04-2018-06-08'
156156
nvidia_command = 'nvidia-smi --query-gpu=index,uuid --format=csv,noheader'
157157
if use_docker:
158158
self.client.images.pull(cuda_image)

0 commit comments

Comments
 (0)