Skip to content

Commit 26744b6

Browse files
committed
update container images to one that has gcloud tools
1 parent b77573d commit 26744b6

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

.github/workflows/pytest_gpu.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ jobs:
2222
python: ["3.10"]
2323

2424
runs-on: "linux-x86-g2-48-l4-4gpu"
25-
container: "gcr.io/tensorflow-testing/nosla-cuda12.3-cudnn9.1-ubuntu20.04-manylinux2014-multipython:latest"
25+
container: "us-central1-docker.pkg.dev/tensorflow-sigs/tensorflow/nosla-cuda12.3-cudnn9.1-ubuntu20.04-manylinux2014-multipython:latest"
2626

2727
name: "Pytest GPU (Build wheels on CUDA 12.3)"
2828
env:
@@ -51,9 +51,9 @@ jobs:
5151
matrix:
5252
test_env: [
5353
{cuda_version: "12.3", runner: "linux-x86-g2-48-l4-4gpu",
54-
image: "gcr.io/tensorflow-testing/nosla-cuda12.3-cudnn9.1-ubuntu20.04-manylinux2014-multipython:latest"},
54+
image: "us-central1-docker.pkg.dev/tensorflow-sigs/tensorflow/nosla-cuda12.3-cudnn9.1-ubuntu20.04-manylinux2014-multipython:latest"},
5555
{cuda_version: "12.1", runner: "linux-x86-g2-48-l4-4gpu",
56-
image: "gcr.io/tensorflow-testing/nosla-cuda12.1-cudnn9.1-ubuntu20.04-manylinux2014-multipython:latest"},
56+
image: "us-central1-docker.pkg.dev/tensorflow-sigs/tensorflow/nosla-cuda12.1-cudnn9.1-ubuntu20.04-manylinux2014-multipython:latest"},
5757
]
5858
python: ["3.10"]
5959

0 commit comments

Comments
 (0)