ray/python/requirements_ml_docker.txt

ipython
# In TF >v2, GPU support is included in the base package.
tensorflow==2.5.0
tensorflow-probability==0.13.0
-f https://download.pytorch.org/whl/torch_stable.html
torch==1.9.0+cu111
-f https://download.pytorch.org/whl/torch_stable.html
torchvision==0.10.0+cu111
[docker] Fix GPU support for tensorflow (#10779) 2020-09-17 10:56:58 -07:00			`ipython`
[Docker] Downgrade to CUDA 11.0 (#17806) 2021-08-13 11:39:06 -07:00			`# In TF >v2, GPU support is included in the base package.`
[RLlib] Bump tf version in ML docker to tf==2.5.0; add tfp to ML-docker. (#18544) 2021-09-15 08:46:37 +02:00			`tensorflow==2.5.0`
			`tensorflow-probability==0.13.0`
[Docker] Use Cuda 11 (#13691) 2021-01-27 13:45:30 -08:00			`-f https://download.pytorch.org/whl/torch_stable.html`
[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			`torch==1.9.0+cu111`
[Docker] Use Cuda 11 (#13691) 2021-01-27 13:45:30 -08:00			`-f https://download.pytorch.org/whl/torch_stable.html`
[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			`torchvision==0.10.0+cu111`