ray/python/requirements_ml_docker.txt

ipython
tensorflow-gpu>=2.4.0
-f https://download.pytorch.org/whl/torch_stable.html
torch==1.7.1+cu110 
-f https://download.pytorch.org/whl/torch_stable.html
torchvision==0.8.2+cu110 
pip; python_version > "3.7"
[docker] Fix GPU support for tensorflow (#10779) 2020-09-17 10:56:58 -07:00			`ipython`
[Docker] Use Cuda 11 (#13691) 2021-01-27 13:45:30 -08:00			`tensorflow-gpu>=2.4.0`
			`-f https://download.pytorch.org/whl/torch_stable.html`
			`torch==1.7.1+cu110`
			`-f https://download.pytorch.org/whl/torch_stable.html`
[docker] Build Python3.6 & Python3.8 Docker Images (#13548) 2021-01-28 15:24:50 -08:00			`torchvision==0.8.2+cu110`
			`pip; python_version > "3.7"`