onnxruntime/tools/ci_build/github/linux/docker/scripts/training/requirements.txt at a77b037e4ad3cf04dd90a1064ff86d2bab7dd0c7 - saymrwulf/onnxruntime - Forgejo: Beyond coding. We forge.

saymrwulf/onnxruntime

mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-02 23:39:58 +00:00

Tianlei Wu c00e13a291

Cherry pick (batch 2) to rel-1.5.1 (#5290 )

* remove implicit linking of tensorrt and dnnl ep shared libs (#5262)
* Update DirectML Nuget to 1.3.0 (#5274)
* Update PyTorch TransformerModel sample (#5275)
* Insert telemetry template into GPU build, add telemry build switches. (#5278)
* Synchronize training dependency versions between Docker image and Python wheel (#5261)
* Downgrade GCC (#5269)
* Remove --enable_symbolic_shape_infer_tests to fix linux ci pipeline build error.

Co-authored-by: Edward Chen
Co-authored-by: George Wu <jywu@microsoft.com>
Co-authored-by: Dwayne Robinson <dwayner@microsoft.com>
Co-authored-by: Thiago Crepaldi <thiago.crepaldi@microsoft.com>
Co-authored-by: Dmitri Smirnov <yuslepukhin@users.noreply.github.com>
Co-authored-by: edgchen1 <18449977+edgchen1@users.noreply.github.com>
Co-authored-by: Changming Sun <chasun@microsoft.com>

2020-09-25 09:26:40 -07:00

8 lines

222 B

Text

Raw Blame History

 --pre
 -f https://download.pytorch.org/whl/nightly/cu102/torch_nightly.html
 # transformers requires sklearn
 sklearn
 transformers==v2.10.0
 torch==1.6.0.dev20200610
 torchvision==0.7.0.dev20200610
 torchtext==0.6.0.dev20200610