Switch RBE image location for CUDA 9 to tensorflow-testing.

PiperOrigin-RevId: 227914044
This commit is contained in:
A. Unique TensorFlower 2019-01-04 14:26:00 -08:00 committed by TensorFlower Gardener
parent 41e333f019
commit dba2134638
4 changed files with 5 additions and 5 deletions

View File

@ -1,7 +1,7 @@
# To push a new version, run:
# $ docker build -f Dockerfile.rbe.cuda9.0-cudnn7-ubuntu14.04 \
# --tag "gcr.io/asci-toolchain/nosla-cuda9.0-cudnn7-ubuntu14.04" .
# $ docker push gcr.io/asci-toolchain/nosla-cuda9.0-cudnn7-ubuntu14.04
# --tag "gcr.io/tensorflow-testing/nosla-cuda9.0-cudnn7-ubuntu14.04" .
# $ docker push gcr.io/tensorflow-testing/nosla-cuda9.0-cudnn7-ubuntu14.04
#
# TODO(klimek): Include clang in this image so we can also target clang
# builds.

View File

@ -32,7 +32,7 @@ platform(
remote_execution_properties = """
properties: {
name: "container-image"
value:"docker://gcr.io/asci-toolchain/nosla-cuda9.0-cudnn7-ubuntu14.04@%s"
value:"docker://gcr.io/tensorflow-testing/nosla-cuda9.0-cudnn7-ubuntu14.04@%s"
}""" % container_digests["cuda9.0-cudnn7-ubuntu14.04"],
)

View File

@ -1,4 +1,4 @@
container_digests = {
"cuda9.0-cudnn7-ubuntu14.04": "sha256:c43ed5341dd765042e0bbd1bf50fadeedd649d1e0c34d81999cb6ce30916cb95",
"cuda9.0-cudnn7-ubuntu14.04": "sha256:006a76ee1838122ff7f21ebac85f24c1ef350d4dd79b3ceff0e4fe649ed90d33",
"cuda10.0-cudnn7-ubuntu14.04": "sha256:e36f05f1ff39e39ddf07122e37f2b1895948bb6f7acc3db37a3c496be5e66228",
}

View File

@ -11,7 +11,7 @@ def _remote_config_workspace():
container_pull(
name = "cuda9.0-cudnn7-ubuntu14.04",
registry = "gcr.io",
repository = "asci-toolchain/nosla-cuda9.0-cudnn7-ubuntu14.04",
repository = "tensorflow-testing/nosla-cuda9.0-cudnn7-ubuntu14.04",
digest = container_digests["cuda9.0-cudnn7-ubuntu14.04"],
)