requirements_ml_docker.txt 278 B

12345678910
  1. ipython
  2. # Needed for Ray Client error message serialization/deserialization.
  3. tblib
  4. # If you make changes to anything below this line, please also change `requirements_dl.txt`!
  5. -f https://download.pytorch.org/whl/torch_stable.html
  6. torch==1.9.0+cu111
  7. torchvision==0.10.0+cu111