ray/python/requirements_ml_docker.txt
Amog Kamsetty 544dff80fa
[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

6 lines
182 B
Text

ipython
tensorflow-gpu>=2.4.0
-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