ray/python/requirements_ml_docker.txt

ipython
# In TF >v2, GPU support is included in the base package.
tensorflow==2.5
-f https://download.pytorch.org/whl/torch_stable.html
torch==1.8.1+cu111
-f https://download.pytorch.org/whl/torch_stable.html
torchvision==0.9.1+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.`
CUDA 11.2 in docker images 2021-08-16 12:31:19 +02:00			`tensorflow==2.5`
[Docker] Use Cuda 11 (#13691) 2021-01-27 13:45:30 -08:00			`-f https://download.pytorch.org/whl/torch_stable.html`
[Docker] Fix torch GPU install on Ray Docker images (#15473) Co-authored-by: Ian Rodney <ian.rodney@gmail.com> 2021-04-26 16:22:25 -07:00			`torch==1.8.1+cu111`
[Docker] Use Cuda 11 (#13691) 2021-01-27 13:45:30 -08:00			`-f https://download.pytorch.org/whl/torch_stable.html`
[Docker] Fix torch GPU install on Ray Docker images (#15473) Co-authored-by: Ian Rodney <ian.rodney@gmail.com> 2021-04-26 16:22:25 -07:00			`torchvision==0.9.1+cu111`