Skip to content

Commit 60baa0c

Browse files
authored
[PyTorch][ARM64][Training] Fix gdrcopy version (#4805)
1 parent 34396ae commit 60baa0c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytorch/training/docker/2.7/py3/cu128/Dockerfile.arm64.gpu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ ARG CUDA_VERSION=12.8.0
66
ARG CUDNN_VERSION=9.8.0.87
77
ARG NCCL_VERSION=2.26.2
88
ARG EFA_VERSION=1.40.0
9-
ARG GDRCOPY_VERSION=2.5.1
9+
ARG GDRCOPY_VERSION=2.5
1010
ARG TE_VERSION=2.0
1111
ARG FLASH_ATTN_VERSION=2.7.3
1212

0 commit comments

Comments
 (0)