Switch RBE image location for CUDA 9 to tensorflow-testing.
PiperOrigin-RevId: 227914044
This commit is contained in:
parent
41e333f019
commit
dba2134638
@ -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.
|
||||
|
2
third_party/toolchains/BUILD
vendored
2
third_party/toolchains/BUILD
vendored
@ -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"],
|
||||
)
|
||||
|
||||
|
@ -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",
|
||||
}
|
||||
|
@ -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"],
|
||||
)
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user