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 = {
"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",
}