使用onnxruntime训练pytorch模型

构建（您需要更新version.txt中的版本号才能上传python whl文件）

rm dist/*
python setup.py bdist_wheel

发布

twine upload dist/*

安装

稳定版

(uninstall onnxruntime version that do not support gpu and training)
pip install torchort onnxruntime=1.9.0+cu111_training

夜间版

(make sure you are not in the ort repo folder - otherwise torchort is taken as already installed)
pip install --pre torchort ort-gpu-nightly-training

(eventually we are aiming at: pip install --pre torchort onnxruntime=1.9.0+cu111_training)

测试

python ./ort/tests/bert_for_sequence_classification.py

在PyTorch训练脚本中使用torchort

import onnxruntime
from torchort import ORTModule
model = ORTModule(model)
# normal PyTorch training script follows