Skip to content

Commit 16c2924

Browse files
authored
ci : update main-cuda.Dockerfile (#3371)
* Update main-cuda.Dockerfile Bump CUDA to 13.0.0 and exclude the `compute_50` arch from build because it was deprecated and now throws an error. * Add quotes in main-cuda.Dockerfile
1 parent 5527454 commit 16c2924

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

.devops/main-cuda.Dockerfile

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
ARG UBUNTU_VERSION=22.04
22
# This needs to generally match the container host's environment.
3-
ARG CUDA_VERSION=12.3.1
3+
ARG CUDA_VERSION=13.0.0
44
# Target the CUDA build image
55
ARG BASE_CUDA_DEV_CONTAINER=nvidia/cuda:${CUDA_VERSION}-devel-ubuntu${UBUNTU_VERSION}
66
# Target the CUDA runtime image
@@ -20,12 +20,12 @@ RUN apt-get update && \
2020
&& rm -rf /var/lib/apt/lists/* /var/cache/apt/archives/*
2121

2222
# Ref: https://stackoverflow.com/a/53464012
23-
ENV CUDA_MAIN_VERSION=12.3
23+
ENV CUDA_MAIN_VERSION=13.0
2424
ENV LD_LIBRARY_PATH /usr/local/cuda-${CUDA_MAIN_VERSION}/compat:$LD_LIBRARY_PATH
2525

2626
COPY .. .
2727
# Enable cuBLAS
28-
RUN make base.en CMAKE_ARGS="-DGGML_CUDA=1"
28+
RUN make base.en CMAKE_ARGS="-DGGML_CUDA=1 -DCMAKE_CUDA_ARCHITECTURES='75;80;86;90'"
2929

3030
RUN find /app/build -name "*.o" -delete && \
3131
find /app/build -name "*.a" -delete && \
@@ -34,7 +34,7 @@ RUN find /app/build -name "*.o" -delete && \
3434
rm -rf /app/build/_deps
3535

3636
FROM ${BASE_CUDA_RUN_CONTAINER} AS runtime
37-
ENV CUDA_MAIN_VERSION=12.3
37+
ENV CUDA_MAIN_VERSION=13.0
3838
ENV LD_LIBRARY_PATH /usr/local/cuda-${CUDA_MAIN_VERSION}/compat:$LD_LIBRARY_PATH
3939
WORKDIR /app
4040

0 commit comments

Comments
 (0)