[ci] Update build-docker-images.py (#31385)

There is no ubuntu20.04 image for cuda 10.2 and 10.1. Signed-off-by: Kornél Csernai <749306+csko@users.noreply.github.com>
ray-project · Jan 6, 2023 · 4feed7f · 4feed7f
1 parent 6c22e59
commit 4feed7f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/ci/build/build-docker-images.py b/ci/build/build-docker-images.py
@@ -43,8 +43,9 @@
     "cu112": "nvidia/cuda:11.2.0-cudnn8-devel-ubuntu20.04",
     "cu111": "nvidia/cuda:11.1.1-cudnn8-devel-ubuntu20.04",
     "cu110": "nvidia/cuda:11.0.3-cudnn8-devel-ubuntu20.04",
-    "cu102": "nvidia/cuda:10.2-cudnn8-devel-ubuntu20.04",
-    "cu101": "nvidia/cuda:10.1-cudnn8-devel-ubuntu20.04",
+    # there is no ubuntu20.04 image for cuda 10.2 and 10.1
+    "cu102": "nvidia/cuda:10.2-cudnn8-devel-ubuntu18.04",
+    "cu101": "nvidia/cuda:10.1-cudnn8-devel-ubuntu18.04",
     "cpu": "ubuntu:focal",
 }