From 7e55a942cd6d06421d9a1f90d52dbfaf19bd4da1 Mon Sep 17 00:00:00 2001 From: Suffian Khan Date: Mon, 13 Dec 2021 20:39:58 -0800 Subject: [PATCH] Add torch 1.10 requirements for rocm (#10028) --- .../stage1/requirements-torch1.10.0_rocm.txt | 12 ++++++++++++ 1 file changed, 12 insertions(+) create mode 100644 tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements-torch1.10.0_rocm.txt diff --git a/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements-torch1.10.0_rocm.txt b/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements-torch1.10.0_rocm.txt new file mode 100644 index 0000000000..1fe2939f03 --- /dev/null +++ b/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements-torch1.10.0_rocm.txt @@ -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