Skip to content

Commit e2c2e7f

Browse files
author
Yadan Wei
committed
fix os_version arg
1 parent ca59323 commit e2c2e7f

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

base/x86_64/gpu/cu128/ubuntu22.04/Dockerfile

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,6 @@ ARG EFA_VERSION="1.40.0"
77
ARG OS_VERSION="ubuntu22.04"
88
FROM nvidia/cuda:12.8.0-base-${OS_VERSION} AS base-builder
99

10-
ARG OS_VERSION
11-
1210
RUN mv /usr/local/cuda/compat /usr/local \
1311
&& apt-get update \
1412
&& apt-get -y upgrade --only-upgrade systemd \
@@ -63,6 +61,7 @@ RUN bash install_python.sh ${PYTHON_VERSION} && rm install_python.sh
6361
FROM base-builder AS cuda-builder
6462
ARG CUDA_MAJOR
6563
ARG CUDA_MINOR
64+
ARG OS_VERSION
6665
COPY install_cuda.sh install_cuda.sh
6766
RUN bash install_cuda.sh "${CUDA_MAJOR}.${CUDA_MINOR}" "${OS_VERSION}" && rm install_cuda.sh
6867

0 commit comments

Comments
 (0)