ray/python/requirements_ml_docker.txt
2021-06-29 11:14:46 -07:00

6 lines
182 B
Text

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