[ROCm] fix python package pipeline (#17136)

ROCm python package pipeline failed because this
PR(https://github.com/microsoft/onnxruntime/pull/16325) changed onnx
version to a commit and we need to build onnx from source. Low protobuf
version will cause build errors.
This PR remove `cmake ` and `protobuf ` from Dockerfile, these two will
install by `install_os_deps.sh`.
This commit is contained in:
PeixuanZuo 2023-08-15 02:22:43 +08:00 committed by GitHub
parent 45f52987a2
commit be2200c00b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -35,12 +35,7 @@ LABEL maintainer="The ManyLinux project"
# of the base docker image. Remove this line moving forward.
RUN yum install -y hipify-clang
# CMake
RUN yum -y install wget
ENV CMAKE_VERSION=3.26.3
RUN cd /usr/local && \
wget -q -O - https://github.com/Kitware/CMake/releases/download/v${CMAKE_VERSION}/cmake-${CMAKE_VERSION}-Linux-x86_64.tar.gz | tar zxf -
ENV PATH=/usr/local/cmake-${CMAKE_VERSION}-linux-x86_64/bin:${PATH}
# rocm lib
RUN yum install -y miopen-hip-devel rocblas-devel rocrand-devel rccl-devel hipsparse-devel hipfft-devel hipcub-devel hipblas-devel rocthrust-devel
@ -100,12 +95,6 @@ RUN export LIBXCRYPT_VERSION=4.4.28 && \
export PERL_DOWNLOAD_URL=https://www.cpan.org/src/5.0 && \
manylinux-entrypoint /build_scripts/install-libxcrypt.sh
COPY scripts/install-protobuf.sh /build_scripts/
RUN export PROTOBUF_VERSION=3.17.3 && \
export PROTOBUF_ROOT=protobuf-all-${PROTOBUF_VERSION} && \
export PROTOBUF_HASH=77ad26d3f65222fd96ccc18b055632b0bfedf295cb748b712a98ba1ac0b704b2 && \
export PROTOBUF_DOWNLOAD_URL=https://github.com/protocolbuffers/protobuf/releases/download/v${PROTOBUF_VERSION} && \
manylinux-entrypoint /build_scripts/install-protobuf.sh
FROM runtime_base AS build_base
COPY build_scripts/install-build-packages.sh /build_scripts/
@ -206,11 +195,6 @@ RUN cd /tmp/scripts && \
/tmp/scripts/install_python_deps.sh -d gpu -p 3.10 $INSTALL_DEPS_EXTRA_ARGS && \
rm -rf /tmp/scripts
# remove protobuf to prevent ambiguity which is used for onnxruntime build
RUN rm -fr /usr/local/bin/protoc \
/usr/local/libproto* \
/usr/local/include/google \
/usr/local/lib/pkgconfig/protobuf*
ARG BUILD_UID=1001
ARG BUILD_USER=onnxruntimedev