torch-mlir/requirements.txt

21 lines
232 B
Plaintext

-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