ray/docker/ray-ml
Amog Kamsetty 84e958f330
[ML] Consolidate and upgrade Deep Learning Dependencies (#18574)
* wip
'

* upgrade requirements

* add file

* fix

* fixes

* Apply suggestions from code review

Try mlagents==0.21.0 for now (works with torch 1.9).

* Apply suggestions from code review

* wip

* wip

* fix

* fix

* upgrade lightning bolts

* address comment

Co-authored-by: Sven Mika <sven@anyscale.io>
2021-09-16 20:16:40 -07:00
..
Dockerfile [ML] Consolidate and upgrade Deep Learning Dependencies (#18574) 2021-09-16 20:16:40 -07:00
README.md [docker] Push to DockerHub in CI (#11442) 2020-10-23 12:02:15 -07:00

About

This image is an extension of the rayproject/ray image. It includes all extended requirements of RLlib, Serve and Tune. It is a well-provisioned starting point for trying out the Ray ecosystem. Find the Dockerfile here.

Tags

  • :latest - The most recent Ray release.
  • :1.x.x - A specific release build.
  • :nightly - The most recent nightly build.
  • :SHA - A specific nightly build.

Suffixes

  • -gpu - These are based off of an NVIDIA CUDA image. They require the Nvidia Docker Runtime to be installed on the host for the container to access GPUs.
  • -cpu- These are based off of an Ubuntu image.
  • Tags without a suffix refer to -cpu images

Other Images