mirror of
https://github.com/vale981/ray
synced 2025-03-04 17:41:43 -05:00
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:
parent
77722b86fd
commit
8692eb6208
1 changed files with 7 additions and 7 deletions
|
@ -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",
|
||||
}
|
||||
|
||||
|
|
Loading…
Add table
Reference in a new issue