File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change 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 :
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
You can’t perform that action at this time.
0 commit comments