Skip to content

Commit a1c684d

Browse files
Update benchmarks.yml
1 parent a4e9a8f commit a1c684d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

.github/workflows/benchmarks.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,7 @@ jobs:
106106
matrix:
107107
job_info:
108108
- os: "linux-x86-g2-48-l4-4gpu"
109-
container: "us-central1-docker.pkg.dev/tensorflow-sigs/tensorflow/ml-build:latest"
109+
image: "gcr.io/tensorflow-testing/nosla-cuda12.3-cudnn9.1-ubuntu20.04-manylinux2014-multipython:latest"
110110
pretty_name: "Linux X86 runner with 4 NVIDIA L4 GPUs"
111111
# Expect more GPU types in the future.
112112
name: ${{ matrix.job_info.pretty_name }}

0 commit comments

Comments
 (0)