mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-15 20:50:42 +00:00
42c176b60c
* update to torch 1.10 * update torchvision version * update torchtext version * remove deprecated option enable_onnx_checker * add unit test to test gradient of GatherElements * add ORTMODULE_ONNX_OPSET_VERSION in a docker file
12 lines
250 B
Text
12 lines
250 B
Text
--pre
|
|
-f https://download.pytorch.org/whl/torch_stable.html
|
|
# transformers requires sklearn
|
|
sklearn
|
|
numpy==1.19.5
|
|
transformers==v2.10.0
|
|
torch==1.10.0+cu113
|
|
torchvision==0.11.1+cu113
|
|
torchtext==0.11.0
|
|
tensorboard>=2.2.0,<2.5.0
|
|
h5py
|
|
setuptools>=41.4.0
|