62 lines
3.0 KiB
Python
62 lines
3.0 KiB
Python
|
"""Docker images used with remote config and RBE."""
|
||
|
|
|
||
|
|
load("//third_party/toolchains/preconfig/generate:containers.bzl", "container_digests")
|
||
|
|
|
||
|
|
containers = {
|
||
|
|
# Built with //tensorflow/tools/ci_build/Dockerfile.rbe.ubuntu16.04-manylinux2010.
|
||
|
|
"ubuntu16.04-manylinux2010": {
|
||
|
|
"registry": "gcr.io",
|
||
|
|
"repository": "tensorflow-testing/nosla-ubuntu16.04-manylinux2010",
|
||
|
|
"digest": container_digests["ubuntu16.04-manylinux2010"],
|
||
|
|
},
|
||
|
|
|
||
|
|
# Built with //tensorflow/tools/ci_build/Dockerfile.rbe.cuda10.0-cudnn7-ubuntu16.04-manylinux2010.
|
||
|
|
"cuda10.0-cudnn7-ubuntu16.04-manylinux2010": {
|
||
|
|
"registry": "gcr.io",
|
||
|
|
"repository": "tensorflow-testing/nosla-cuda10.0-cudnn7-ubuntu16.04-manylinux2010",
|
||
|
|
"digest": container_digests["cuda10.0-cudnn7-ubuntu16.04-manylinux2010"],
|
||
|
|
},
|
||
|
|
|
||
|
|
# Built with //tensorflow/tools/ci_build/Dockerfile.rbe.cuda10.1-cudnn7-ubuntu16.04-manylinux2010.
|
||
|
|
"cuda10.1-cudnn7-ubuntu16.04-manylinux2010": {
|
||
|
|
"registry": "gcr.io",
|
||
|
|
"repository": "tensorflow-testing/nosla-cuda10.1-cudnn7-ubuntu16.04-manylinux2010",
|
||
|
|
"digest": container_digests["cuda10.1-cudnn7-ubuntu16.04-manylinux2010"],
|
||
|
|
},
|
||
|
|
|
||
|
|
# Built with //tensorflow/tools/ci_build/Dockerfile.rbe.cuda10.1-cudnn7-ubuntu16.04-manylinux2010-multipython.
|
||
|
|
"cuda10.1-cudnn7-ubuntu16.04-manylinux2010-multipython": {
|
||
|
|
"registry": "gcr.io",
|
||
|
|
"repository": "tensorflow-testing/nosla-cuda10.1-cudnn7-ubuntu16.04-manylinux2010-multipython",
|
||
|
|
"digest": container_digests["cuda10.1-cudnn7-ubuntu16.04-manylinux2010-multipython"],
|
||
|
|
},
|
||
|
|
|
||
|
|
# Built with //tensorflow/tools/ci_build/Dockerfile.rbe.cuda10.1-cudnn7-ubuntu18.04-manylinux2010-multipython.
|
||
|
|
"cuda10.1-cudnn7-ubuntu18.04-manylinux2010-multipython": {
|
||
|
|
"registry": "gcr.io",
|
||
|
|
"repository": "tensorflow-testing/nosla-cuda10.1-cudnn7-ubuntu18.04-manylinux2010-multipython",
|
||
|
|
"digest": container_digests["cuda10.1-cudnn7-ubuntu18.04-manylinux2010-multipython"],
|
||
|
|
},
|
||
|
|
|
||
|
|
# Built with //tensorflow/tools/ci_build/Dockerfile.rbe.cuda11.0-cudnn8-ubuntu18.04-manylinux2010-multipython.
|
||
|
|
"cuda11.0-cudnn8-ubuntu18.04-manylinux2010-multipython": {
|
||
|
|
"registry": "gcr.io",
|
||
|
|
"repository": "tensorflow-testing/nosla-cuda11.0-cudnn8-ubuntu18.04-manylinux2010-multipython",
|
||
|
|
"digest": container_digests["cuda11.0-cudnn8-ubuntu18.04-manylinux2010-multipython"],
|
||
|
|
},
|
||
|
|
|
||
|
|
# Built with //tensorflow/tools/ci_build/Dockerfile.rbe.rocm-ubuntu18.04-manylinux2010-multipython.
|
||
|
|
"rocm-ubuntu18.04-manylinux2010-multipython": {
|
||
|
|
"registry": "gcr.io",
|
||
|
|
"repository": "tensorflow-testing/nosla-rocm-ubuntu18.04-manylinux2010-multipython",
|
||
|
|
"digest": container_digests["rocm-ubuntu18.04-manylinux2010-multipython"],
|
||
|
|
},
|
||
|
|
|
||
|
|
# Built by gunan@ from a private Dockerfile.
|
||
|
|
"windows-1803": {
|
||
|
|
"registry": "gcr.io",
|
||
|
|
"repository": "tensorflow-testing/tf-win-rbe",
|
||
|
|
"digest": container_digests["windows-1803"],
|
||
|
|
},
|
||
|
|
}
|