From 8692eb620878cfe214fb670df5d4c1bf83ab6a51 Mon Sep 17 00:00:00 2001 From: Amog Kamsetty Date: Fri, 2 Sep 2022 02:30:11 -0700 Subject: [PATCH] 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 --- ci/build/build-docker-images.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/ci/build/build-docker-images.py b/ci/build/build-docker-images.py index 131f018f0..b5fc81ed3 100644 --- a/ci/build/build-docker-images.py +++ b/ci/build/build-docker-images.py @@ -38,13 +38,13 @@ PY_MATRIX = { } 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", }