mirror of
https://github.com/vale981/ray
synced 2025-03-05 10:01: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 = {
|
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",
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Add table
Reference in a new issue