@@ -4,7 +4,7 @@ ARG CUDA_VERSION
44ARG NCCL_VERSION
55ARG MINIFORGE_VERSION=24.9.2-0
66ARG CMAKE_VERSION=3.31.2
7- ARG MAVEN_VERSION=3.9.15
7+ ARG MAVEN_VERSION=3.9.16
88
99SHELL ["/bin/bash", "-c"]
1010
@@ -14,13 +14,11 @@ ENV CXX=/opt/rh/gcc-toolset-10/root/usr/bin/c++
1414ENV CPP=/opt/rh/gcc-toolset-10/root/usr/bin/cpp
1515ENV GOSU_VERSION=1.10
1616
17+ COPY scripts/setup_cuda_rockylinux.sh scripts/install_gosu.sh /scripts/
18+
1719# Install all basic requirements
1820RUN \
19- curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/rhel8/x86_64/D42D0685.pub | sed '/^Version/d' \
20- > /etc/pki/rpm-gpg/RPM-GPG-KEY-NVIDIA && \
21- dnf -y update && \
22- dnf -y install dnf-plugins-core && \
23- dnf config-manager --set-enabled powertools && \
21+ bash /scripts/setup_cuda_rockylinux.sh && \
2422 dnf install -y tar unzip wget xz git which ninja-build java-17-openjdk-devel gcc-toolset-10-gcc gcc-toolset-10-binutils gcc-toolset-10-gcc-c++ && \
2523 # Miniforge
2624 wget -nv -O conda.sh https://github.com/conda-forge/miniforge/releases/download/$MINIFORGE_VERSION/Miniforge3-$MINIFORGE_VERSION-Linux-x86_64.sh && \
4543RUN pip install numpy pytest scipy scikit-learn wheel kubernetes awscli
4644
4745# Install lightweight sudo (not bound to TTY)
48- RUN set -ex; \
49- wget -nv -nc -O /usr/local/bin/gosu "https://github.com/tianon/gosu/releases/download/$GOSU_VERSION/gosu-amd64" && \
50- chmod +x /usr/local/bin/gosu && \
51- gosu nobody true
46+ RUN sh /scripts/install_gosu.sh
5247
5348# Default entry-point to use if running locally
5449# It will preserve attributes of created files
0 commit comments