onnxruntime/requirements-training.txt

cerberus
flatbuffers
h5py
numpy >= 1.16.6
onnx
packaging
protobuf
sympy
setuptools>=61.0.0
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-24 01:43:19 +00:00			`cerberus`
			`flatbuffers`
			`h5py`
			`numpy >= 1.16.6`
			`onnx`
			`packaging`
Remove protobuf pin from training requirements (#13695) 2022-11-22 20:27:18 +00:00			`protobuf`
Add post-install command to build PyTorch CPP extensions from within onnxruntime package (#8027) ORTModule requires two PyTorch CPP extensions that are currently JIT compiled. The runtime compilation can cause issues in some environments without all build requirements or in environments with multiple instances of ORTModule running in parallel This PR creates a custom command to compile such extensions that must be manually executed before ORTModule is executed for the first time. When users try to use ORTModule before the extensions are compiled, an error with instructions are raised PyTorch CPP Extensions for ORTModule can be compiled by running: python -m onnxruntime.training.ortmodule.torch_cpp_extensions.install Full build environment is needed for this 2021-06-29 01:11:58 +00:00			`sympy`
ONNX 1.15 integration (#17125) ### Description this is for ORT 1.17.0 - make ORT to use ONNX release 1.15.0 branch. Eventually will update to the release tag once ONNX 1.15.0 is released ### Motivation and Context Prepare for ORT 1.17.0 release. People can start work on new and updated ONNX ops in ORT. --------- Signed-off-by: Liqun Fu <liqfu@microsoft.com> 2023-09-26 21:44:48 +00:00			`setuptools>=61.0.0`