Skip to content

Commit 8abb71d

Browse files
authored
🔥 remove cuda-runtime entirely (#60)
Previous changes rebased the release stage onto `base` instead of `cuda-runtime`. This PR removes the `cuda-runtime` stage entirely. Signed-off-by: Joe Runde <[email protected]>
1 parent 4b545ad commit 8abb71d

File tree

1 file changed

+1
-20
lines changed

1 file changed

+1
-20
lines changed

Dockerfile

Lines changed: 1 addition & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -56,25 +56,6 @@ ENV CUDA_HOME="/usr/local/cuda" \
5656
LD_LIBRARY_PATH="/usr/local/nvidia/lib:/usr/local/nvidia/lib64:$CUDA_HOME/lib64:$CUDA_HOME/extras/CUPTI/lib64:${LD_LIBRARY_PATH}"
5757

5858

59-
## CUDA Runtime ################################################################
60-
FROM cuda-base as cuda-runtime
61-
62-
ENV NV_NVTX_VERSION=11.8.86-1 \
63-
NV_LIBNPP_VERSION=11.8.0.86-1 \
64-
NV_LIBCUBLAS_VERSION=11.11.3.6-1 \
65-
NV_LIBNCCL_PACKAGE_VERSION=2.15.5-1+cuda11.8
66-
67-
RUN dnf config-manager \
68-
--add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel9/x86_64/cuda-rhel9.repo \
69-
&& dnf install -y \
70-
cuda-libraries-11-8-${NV_CUDA_LIB_VERSION} \
71-
cuda-nvtx-11-8-${NV_NVTX_VERSION} \
72-
libnpp-11-8-${NV_LIBNPP_VERSION} \
73-
libcublas-11-8-${NV_LIBCUBLAS_VERSION} \
74-
libnccl-${NV_LIBNCCL_PACKAGE_VERSION} \
75-
&& dnf clean all
76-
77-
7859
## CUDA Development ############################################################
7960
FROM cuda-base as cuda-devel
8061

@@ -269,7 +250,7 @@ COPY --from=auto-gptq-installer /usr/src/auto-gptq-wheel /usr/src/auto-gptq-whee
269250

270251
## Full set of python installations for server release #########################
271252

272-
FROM cuda-runtime as python-installations
253+
FROM python-builder as python-installations
273254

274255
ARG PYTHON_VERSION
275256
ARG SITE_PACKAGES=/opt/tgis/lib/python${PYTHON_VERSION}/site-packages

0 commit comments

Comments
 (0)