mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-16 21:00:14 +00:00
773992c7d4
* add tensor board, remove torch.distributed.lanuch because ort nccl depends on MPI. Use MPI to launch parallel training. Co-authored-by: liqun <liqun@OrtTrainingDev4.af05slrtruoetgaxwwjv5nsq5e.px.internal.cloudapp.net>
10 lines
247 B
Text
10 lines
247 B
Text
--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
|
|
tensorboard==v2.0.0
|
|
h5py
|