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

test migration to rocm4.2 (#7800)

Browse source
This commit is contained in:
Suffian Khan 2021-05-24 11:48:44 -07:00 committed by GitHub
parent da5ab325ee
commit 02c78a8aa8
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tools/ci_build/github/azure-pipelines/templates/py-packaging-stage.yml
View file

@ -285,7 +285,7 @@ stages:
/onnxruntime_src/tools/ci_build/build.py \
--config Release \
--use_rocm \
--rocm_version=4.1 \
--rocm_version=4.2 \
--rocm_home=/opt/rocm \
--nccl_home=/opt/rocm \
--update \

2
tools/ci_build/github/linux/docker/Dockerfile.manylinux2014_rocm
View file

@ -1,4 +1,4 @@
FROM rocm/pytorch:rocm4.1.1_centos7_py3.6_pytorch
FROM rocm/pytorch:rocm4.2_centos7_py3.6_pytorch
#Build manylinux2014 docker image begin
ENV AUDITWHEEL_ARCH x86_64

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

@ -58,7 +58,7 @@ if [ $DEVICE_TYPE = "gpu" ]; then
${PYTHON_EXE} -m pip install -r ${0/%install_python_deps.sh/training\/ortmodule\/stage2\/requirements.txt}
else
${PYTHON_EXE} -m pip install \
--pre -f https://download.pytorch.org/whl/nightly/rocm4.1/torch_nightly.html \
--pre -f https://download.pytorch.org/whl/nightly/rocm4.2/torch_nightly.html \
torch torchvision torchtext
${PYTHON_EXE} -m pip install -r ${0/%install_python_deps.sh/training\/ortmodule\/stage1\/requirements-rocm.txt}
${PYTHON_EXE} -m pip install fairscale