2021-09-16 20:16:40 -07:00
|
|
|
# These requirements are used for the CI and CPU-only Docker images so we install CPU only versions of torch.
|
|
|
|
|
# For GPU Docker images, you should install requirements_ml_docker.txt instead.
|
|
|
|
|
|
2021-11-05 00:58:34 -07:00
|
|
|
|
|
|
|
|
# If you make changes to this file, please also change `requirements_ml_docker.txt`!
|
|
|
|
|
|
|
|
|
|
|
2021-11-03 15:32:10 -07:00
|
|
|
# TODO(amogkam): Remove after https://github.com/tensorflow/tensorflow/issues/52922 is fixed.
|
|
|
|
|
keras==2.6.0
|
2021-10-18 22:29:16 +01:00
|
|
|
tensorflow==2.6.0
|
|
|
|
|
tensorflow-probability==0.14.0
|
2021-09-16 20:16:40 -07:00
|
|
|
|
|
|
|
|
torch==1.9.0;sys_platform=="darwin"
|
|
|
|
|
torchvision==0.10.0;sys_platform=="darwin"
|
|
|
|
|
|
|
|
|
|
# On non-OSX machines only install CPU version of torch and torchvision
|
|
|
|
|
-f https://download.pytorch.org/whl/torch_stable.html
|
|
|
|
|
torch==1.9.0+cpu;sys_platform!="darwin"
|
|
|
|
|
-f https://download.pytorch.org/whl/torch_stable.html
|
|
|
|
|
torchvision==0.10.0+cpu;sys_platform!="darwin"
|
