requirements.txt 634 B

1234567891011121314151617181920212223242526272829
  1. # On Ubuntu, install the following packages:
  2. # - python3
  3. # - python3-venv
  4. #
  5. # And then the virtual environment can be created by running:
  6. #
  7. # python3 -m venv .venv
  8. # .venv/bin/pip install --upgrade pip
  9. # .venv/bin/pip install -r requirements.txt
  10. #
  11. # Then use .venv/bin/python to run the various training scripts.
  12. blosc>=1.8.1,<2
  13. diskcache==4.1.0
  14. h5py
  15. jupyter
  16. matplotlib
  17. numpy<=1.16.4
  18. Pillow
  19. scikit-image>=0.15.0
  20. scipy
  21. SimpleITK
  22. tensorflow>=1.14,<2
  23. # The book was written targeting PyTorch 1.4, but everything in it should work fine with more up-to-date versions.
  24. torch==1.4
  25. # future is here due to a missing dep in caffe2
  26. # future