update linux ci

microsoft · Dec 17, 2024 · 0cdb4ac · 0cdb4ac
1 parent db3062b
commit 0cdb4ac
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/tools/ci_build/github/linux/docker/manylinux/Dockerfile.manylinux2_28_cpu b/tools/ci_build/github/linux/docker/manylinux/Dockerfile.manylinux2_28_cpu
@@ -2,7 +2,9 @@ FROM onnxruntimebuildcache.azurecr.io/internal/azureml/onnxruntime/build/cpu_x64
 
 ADD scripts /tmp/scripts
 RUN cd /tmp/scripts && /tmp/scripts/install_centos_gcc12.sh && /tmp/scripts/install_deps.sh && rm -rf /tmp/scripts
-
+ENV PATH="/usr/.cargo/bin:$PATH"
+ENV RUSTUP_HOME="/usr/.rustup"
+ENV CARGO_HOME="/usr/.cargo"
 ARG BUILD_UID=1001
 ARG BUILD_USER=onnxruntimedev
 RUN adduser --uid $BUILD_UID $BUILD_USER

diff --git a/tools/ci_build/github/linux/docker/manylinux/Dockerfile.manylinux2_28_cuda_11.8 b/tools/ci_build/github/linux/docker/manylinux/Dockerfile.manylinux2_28_cuda_11.8
@@ -10,7 +10,9 @@ else \
     echo "Using default gcc because CUDA version is less than 12"; \
     cd /tmp/scripts && /tmp/scripts/install_centos.sh && /tmp/scripts/install_deps.sh && rm -rf /tmp/scripts; \
 fi
-
+ENV PATH="/usr/.cargo/bin:$PATH"
+ENV RUSTUP_HOME="/usr/.rustup"
+ENV CARGO_HOME="/usr/.cargo"
 ARG BUILD_UID=1001
 ARG BUILD_USER=onnxruntimedev
 RUN adduser --uid $BUILD_UID $BUILD_USER

diff --git a/tools/ci_build/github/linux/docker/manylinux/Dockerfile.manylinux2_28_rocm b/tools/ci_build/github/linux/docker/manylinux/Dockerfile.manylinux2_28_rocm
@@ -2,7 +2,9 @@ FROM onnxruntimebuildcache.azurecr.io/internal/azureml/onnxruntime/build/cpu_x64
 
 ADD scripts /tmp/scripts
 RUN cd /tmp/scripts && /tmp/scripts/install_centos_gcc12.sh && /tmp/scripts/install_deps.sh && rm -rf /tmp/scripts
-
+ENV PATH="/usr/.cargo/bin:$PATH"
+ENV RUSTUP_HOME="/usr/.rustup"
+ENV CARGO_HOME="/usr/.cargo"
 ARG BUILD_UID=1001
 ARG BUILD_USER=onnxruntimedev
 RUN adduser --uid $BUILD_UID $BUILD_USER