We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 34396ae commit 60baa0cCopy full SHA for 60baa0c
pytorch/training/docker/2.7/py3/cu128/Dockerfile.arm64.gpu
@@ -6,7 +6,7 @@ ARG CUDA_VERSION=12.8.0
6
ARG CUDNN_VERSION=9.8.0.87
7
ARG NCCL_VERSION=2.26.2
8
ARG EFA_VERSION=1.40.0
9
-ARG GDRCOPY_VERSION=2.5.1
+ARG GDRCOPY_VERSION=2.5
10
ARG TE_VERSION=2.0
11
ARG FLASH_ATTN_VERSION=2.7.3
12
0 commit comments