onnxruntime/requirements-training.txt
KeDengMS 6987106bf5
Add missing Python dependencies for ORT training (#7104)
* Add missing Python dependencies for training

cerberus - option parsing
h5py - checkpoint
onnx - model proto
packaging/sympy - symbolic shape inference

* Separate requirements.txt for inference and training Python packages.
2021-03-23 18:43:19 -07:00

8 lines
No EOL
71 B
Text

cerberus
flatbuffers
h5py
numpy >= 1.16.6
onnx
packaging
protobuf
sympy