Skip to content

Commit 4322c0a

Browse files
authored
Remove epel-release package from wheel Dockerfiles (#2987)
Remove epel-release package from wheel Dockerfiles Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
1 parent 76c2a9e commit 4322c0a

2 files changed

Lines changed: 1 addition & 3 deletions

File tree

build_tools/wheel_utils/Dockerfile.aarch

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,6 @@ ENV CUDA_MAJOR=${CUDA_MAJOR}
2323

2424
# Cuda toolkit, cudnn, driver.
2525
RUN dnf config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel8/sbsa/cuda-rhel8.repo
26-
RUN dnf -y install epel-release
2726
RUN dnf -y install cuda-compiler-${CUDA_MAJOR}-${CUDA_MINOR}.aarch64 \
2827
cuda-libraries-${CUDA_MAJOR}-${CUDA_MINOR}.aarch64 \
2928
cuda-libraries-devel-${CUDA_MAJOR}-${CUDA_MINOR}.aarch64

build_tools/wheel_utils/Dockerfile.x86

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,6 @@ ENV CUDA_MAJOR=${CUDA_MAJOR}
2323

2424
# Cuda toolkit, cudnn, driver.
2525
RUN dnf config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel8/x86_64/cuda-rhel8.repo
26-
RUN dnf -y install epel-release
2726
RUN dnf -y install cuda-compiler-${CUDA_MAJOR}-${CUDA_MINOR}.x86_64 \
2827
cuda-libraries-${CUDA_MAJOR}-${CUDA_MINOR}.x86_64 \
2928
cuda-libraries-devel-${CUDA_MAJOR}-${CUDA_MINOR}.x86_64
@@ -44,4 +43,4 @@ ENV CUDA_PATH=/usr/local/cuda
4443
ENV CUDADIR=/usr/local/cuda
4544
ENV NVTE_RELEASE_BUILD=1
4645

47-
CMD ["/bin/bash", "-c", "bash /TransformerEngine/build_tools/wheel_utils/build_wheels.sh manylinux_2_28_x86_64 $BUILD_METAPACKAGE $BUILD_COMMON $BUILD_PYTORCH $BUILD_JAX $CUDA_MAJOR"]
46+
CMD ["/bin/bash", "-c", "bash /TransformerEngine/build_tools/wheel_utils/build_wheels.sh manylinux_2_28_x86_64 $BUILD_METAPACKAGE $BUILD_COMMON $BUILD_PYTORCH $BUILD_JAX $CUDA_MAJOR"]

0 commit comments

Comments
 (0)