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>
This commit is contained in:
Amog Kamsetty 2022-09-02 02:30:11 -07:00 committed by GitHub
parent 77722b86fd
commit 8692eb6208
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -38,13 +38,13 @@ PY_MATRIX = {
} }
BASE_IMAGES = { BASE_IMAGES = {
"cu116": "nvidia/cuda:11.6.1-cudnn8-devel-ubuntu18.04", "cu116": "nvidia/cuda:11.6.1-cudnn8-devel-ubuntu20.04",
"cu113": "nvidia/cuda:11.3.1-cudnn8-devel-ubuntu18.04", "cu113": "nvidia/cuda:11.3.1-cudnn8-devel-ubuntu20.04",
"cu112": "nvidia/cuda:11.2.0-cudnn8-devel-ubuntu18.04", "cu112": "nvidia/cuda:11.2.0-cudnn8-devel-ubuntu20.04",
"cu111": "nvidia/cuda:11.1.1-cudnn8-devel-ubuntu18.04", "cu111": "nvidia/cuda:11.1.1-cudnn8-devel-ubuntu20.04",
"cu110": "nvidia/cuda:11.0.3-cudnn8-devel-ubuntu18.04", "cu110": "nvidia/cuda:11.0.3-cudnn8-devel-ubuntu20.04",
"cu102": "nvidia/cuda:10.2-cudnn8-devel-ubuntu18.04", "cu102": "nvidia/cuda:10.2-cudnn8-devel-ubuntu20.04",
"cu101": "nvidia/cuda:10.1-cudnn8-devel-ubuntu18.04", "cu101": "nvidia/cuda:10.1-cudnn8-devel-ubuntu20.04",
"cpu": "ubuntu:focal", "cpu": "ubuntu:focal",
} }