60 lines
2.8 KiB
Python
60 lines
2.8 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-ubuntu16.04
|
|
"rocm-ubuntu16.04": {
|
|
"registry": "gcr.io",
|
|
"repository": "tensorflow-testing/nosla-rocm-ubuntu16.04",
|
|
"digest": container_digests["rocm-ubuntu16.04"],
|
|
},
|
|
"windows-1803": {
|
|
"registry": "gcr.io",
|
|
"repository": "tensorflow-testing/tf-win-rbe",
|
|
"digest": container_digests["windows-1803"],
|
|
},
|
|
}
|