torch-mlir/requirements.txt

17 lines
187 B
Plaintext

-f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
--pre
torch
numpy
torchvision
# Build requirements.
pybind11
wheel
setuptools
cmake
ninja
# Test Requirements
pillow