ray/python/requirements_ml_docker.txt
2021-01-27 13:45:30 -08:00

6 lines
No EOL
183 B
Text

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