saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-29 20:14:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add torch 1.10 requirements for rocm (#10028)

Browse source
This commit is contained in:
Suffian Khan 2021-12-13 20:39:58 -08:00 committed by GitHub
parent 6de2a878cb
commit 7e55a942cd
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements-torch1.10.0_rocm.txt Normal file
View file

@ -0,0 +1,12 @@
# transformers requires sklearn
--pre
-f https://download.pytorch.org/whl/rocm4.2/torch_stable.html
torch==1.10.0
pandas
sklearn
numpy==1.19.5
transformers==v4.3.2
tensorboard>=2.2.0,<2.5.0
h5py
wget
setuptools>=41.4.0