Update build-docker-images.py (#28249)

Upgrade to a more recent Ubuntu version based on user feedback and also matching the version we use for testing in CI. Signed-off-by: Amog Kamsetty <amogkamsetty@yahoo.com>
ray-project · Sep 2, 2022 · 8692eb6 · 8692eb6
1 parent 77722b8
commit 8692eb6
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/ci/build/build-docker-images.py b/ci/build/build-docker-images.py
@@ -38,13 +38,13 @@
 }
 
 BASE_IMAGES = {
-    "cu116": "nvidia/cuda:11.6.1-cudnn8-devel-ubuntu18.04",
-    "cu113": "nvidia/cuda:11.3.1-cudnn8-devel-ubuntu18.04",
-    "cu112": "nvidia/cuda:11.2.0-cudnn8-devel-ubuntu18.04",
-    "cu111": "nvidia/cuda:11.1.1-cudnn8-devel-ubuntu18.04",
-    "cu110": "nvidia/cuda:11.0.3-cudnn8-devel-ubuntu18.04",
-    "cu102": "nvidia/cuda:10.2-cudnn8-devel-ubuntu18.04",
-    "cu101": "nvidia/cuda:10.1-cudnn8-devel-ubuntu18.04",
+    "cu116": "nvidia/cuda:11.6.1-cudnn8-devel-ubuntu20.04",
+    "cu113": "nvidia/cuda:11.3.1-cudnn8-devel-ubuntu20.04",
+    "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",
     "cpu": "ubuntu:focal",
 }