Skip to content

Commit 3d418d9

Browse files
authored
Update NCCL Jobset image in GKE A4X Max (#6007)
1 parent 22d74d7 commit 3d418d9

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

examples/gke-a4x-max-bm/nccl-jobset-example.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ spec:
7878
- name: nccl-test
7979
stdin: true
8080
tty: true
81-
image: us-docker.pkg.dev/gce-ai-infra/gpudirect-gib/nccl-gib-a4x-max-arm64:v1.1.1
81+
image: us-docker.pkg.dev/gce-ai-infra/gpudirect-gib/nccl-plugin-gib-diagnostic-arm64:v1.1.2
8282
securityContext:
8383
capabilities:
8484
add: ["IPC_LOCK"]

0 commit comments

Comments
 (0)