Skip to content

Commit d2632ed

Browse files
committed
add back libcudnn9-headers-cuda-12
1 parent b5cde70 commit d2632ed

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

pytorch/training/docker/2.8/py3/cu129/Dockerfile.gpu

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -105,6 +105,7 @@ RUN apt-get update \
105105
cuda-toolkit-12=${CUDA_VERSION}-1 \
106106
libcudnn9-cuda-12=${CUDNN_VERSION}-1 \
107107
libcudnn9-dev-cuda-12=${CUDNN_VERSION}-1 \
108+
libcudnn9-headers-cuda-12=${CUDNN_VERSION}-1 \
108109
libhwloc-dev \
109110
libgomp1 \
110111
libibverbs-dev \

0 commit comments

Comments
 (0)