-r pytorch-requirements.txt

numpy
torchvision

# Build requirements.
pybind11
wheel
setuptools
cmake
ninja

# Workaround for what should be a torch dep
# See discussion in #1174
pyyaml

# Test Requirements
pillow
dill
multiprocess