ray/docker
Amog Kamsetty aa8a7dcb48
[Docker] Add Cuda 11.6 support (#26695)
Signed-off-by: Amog Kamsetty amogkamsetty@yahoo.com

Latest Pytorch version has wheels for CUDA 11.6. Per user request, adding a 11.6 image as part of our build pipeline.
2022-07-26 10:15:53 -07:00
..
autoscaler [Doc] update docker readme files to include Python versions (#25099) 2022-05-25 19:42:24 -07:00
base-deps [Core] install jemalloc in Ray docker and use jemalloc in benchmark release tests (#26112) 2022-06-27 23:26:56 -07:00
development [ci] Clean up ci/ directory (refactor ci/travis) (#23866) 2022-04-13 18:11:30 +01:00
examples [RLlib] Upgrade gym version to 0.21 and deprecate pendulum-v0. (#19535) 2021-11-03 16:24:00 +01:00
ray [Doc] update docker readme files to include Python versions (#25099) 2022-05-25 19:42:24 -07:00
ray-deps Fix redis dependency (#26459) 2022-07-12 16:07:09 -07:00
ray-ml [Doc] update docker readme files to include Python versions (#25099) 2022-05-25 19:42:24 -07:00
ray-worker-container remove unmaintained blist (#23957) 2022-04-17 16:06:04 -07:00
retag-lambda [Docker] Add Cuda 11.6 support (#26695) 2022-07-26 10:15:53 -07:00
fix-docker-latest.sh [Docker] Update echo in fix-docker-latest.sh (#22123) 2022-02-07 08:50:36 -08:00
README.md [CI] Fix-Up Docker Build (Use Python) (#11139) 2020-10-12 14:22:51 -07:00

Overview of how the ray images are built:

Images without a "-cpu" or "-gpu" tag are built on ubuntu/focal. They are just an alias for -cpu (e.g. ray:latest is the same as ray:latest-cpu).

ubuntu/focal
└── base-deps:cpu
    └── ray-deps:cpu
        └── ray:cpu
            └── ray-ml:cpu

nvidia/cuda
└── base-deps:gpu
    └── ray-deps:gpu
        └── ray:gpu
            └── ray-ml:gpu