Skip to content

Commit a40a017

Browse files
Pin libcudnn9 dependency for CUDA 12
1 parent c4ba19f commit a40a017

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

docker/Dockerfile.base

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -317,7 +317,7 @@ RUN --mount=type=cache,target=/var/cache/apt \
317317
cuda-sanitizer-12-6 \
318318
cuda-toolkit-12-6 \
319319
libcublas-12-6 \
320-
libcudnn9 \
320+
libcudnn9-cuda-12 \
321321
libcusparse-12-6 \
322322
libnpp-12-6
323323

0 commit comments

Comments
 (0)