saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-26 22:35:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update Torch Version to Fix ATen CPU Pipeline Failure (#20845)

Browse source 
Update Torch Version to Fix ATen CPU Pipeline Failure.
This commit is contained in:
Vincent Wang 2024-05-29 16:04:18 +08:00 committed by GitHub
parent 3044aa8743
commit e77f238dc6
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tools/ci_build/github/linux/docker/scripts/manylinux/install_deps_aten.sh
View file

@ -43,7 +43,7 @@ do
if ! [[ ${PYTHON_EXE} = "/opt/python/cp310-cp310/bin/python3.10" ]]; then
${PYTHON_EXE} -m pip install -r ${0/%install_deps_aten\.sh/..\/training\/ortmodule\/stage1\/requirements_torch_cpu\/requirements.txt}
else
${PYTHON_EXE} -m pip install torch==2.0.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
${PYTHON_EXE} -m pip install torch==2.3.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
fi
done

2
tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch_cpu/requirements.txt
View file

@ -1,3 +1,3 @@
-f https://download.pytorch.org/whl/torch_stable.html
torch==2.0.0+cpu
torch==2.3.0+cpu
setuptools>=68.2.2