torch-mlir/requirements.txt

21 lines
265 B
Plaintext
Raw Normal View History

-f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
--pre
torch
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