We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent ca59323 commit e2c2e7fCopy full SHA for e2c2e7f
base/x86_64/gpu/cu128/ubuntu22.04/Dockerfile
@@ -7,8 +7,6 @@ ARG EFA_VERSION="1.40.0"
7
ARG OS_VERSION="ubuntu22.04"
8
FROM nvidia/cuda:12.8.0-base-${OS_VERSION} AS base-builder
9
10
-ARG OS_VERSION
11
-
12
RUN mv /usr/local/cuda/compat /usr/local \
13
&& apt-get update \
14
&& apt-get -y upgrade --only-upgrade systemd \
@@ -63,6 +61,7 @@ RUN bash install_python.sh ${PYTHON_VERSION} && rm install_python.sh
63
61
FROM base-builder AS cuda-builder
64
62
ARG CUDA_MAJOR
65
ARG CUDA_MINOR
+ARG OS_VERSION
66
COPY install_cuda.sh install_cuda.sh
67
RUN bash install_cuda.sh "${CUDA_MAJOR}.${CUDA_MINOR}" "${OS_VERSION}" && rm install_cuda.sh
68
0 commit comments