ci: cuda/cudnn version upgrade (#60)

kibae · Jun 28, 2024 · 75e16c5 · 75e16c5
1 parent 97f2e23
commit 75e16c5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/deploy/build-docker/linux-cuda12.dockerfile b/deploy/build-docker/linux-cuda12.dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.2.2-cudnn8-devel-ubuntu22.04 AS builder
+FROM nvidia/cuda:12.3.2-cudnn9-devel-ubuntu22.04 AS builder
 
 RUN apt update && apt install -y curl wget git build-essential cmake pkg-config libboost-all-dev libssl-dev
 RUN mkdir -p /app/source
@@ -20,7 +20,7 @@ RUN cmake --build build --parallel 4 --target onnxruntime_server_standalone
 RUN cmake --install build --prefix /app/onnxruntime-server
 
 # target
-FROM nvidia/cuda:12.2.2-cudnn8-runtime-ubuntu22.04 AS target
+FROM nvidia/cuda:12.3.2-cudnn9-runtime-ubuntu22.04 AS target
 COPY --from=builder /app/onnxruntime-server /app
 COPY --from=builder /usr/local/onnxruntime /usr/local/onnxruntime