xref: /aosp_15_r20/external/tensorflow/tensorflow/tools/toolchains/remote_config/containers.bzl (revision b6fb3261f9314811a0f4371741dbb8839866f948) 
1"""Docker images used with remote config and RBE."""
2
3"""SHA 256 values for each image."""
4container_digests = {
5    # TF now uses only this container
6    "cuda11.2-cudnn8.1-ubuntu20.04-manylinux2014-multipython": "sha256:48612bd85709cd014711d0b0f87e0806f3567d06d2e81c6e860516b87498b821",
7    # JAX manylinux2014 configs.
8    "cuda11.1-cudnn8-ubuntu20.04-manylinux2014-multipython": "sha256:3764b49e64a16e5778995fc21a119ff0e364174ecbf461f741701b48e6d4f204",
9    "cuda11.4-cudnn8.2-ubuntu20.04-manylinux2014-multipython": "sha256:6531a7ed4b1524e9f997ad10ad214af79e1042bc7ec6efe2f3e0692bafdb968f",
10    # ROCM, probably not all of them still in use
11    "rocm-ubuntu18.04-manylinux2010-multipython": "sha256:6e953a09b145df338bcb03e9e36f99b291140c29b72d0a048fb6c5905ccad5eb",
12    "rocm-ubuntu20.04-manylinux2014-multipython": "sha256:26720ebae4d6d12b1fca529616bfacfd0460990d4725af35e0f4af3c2422f227",
13    # Unused?
14    "ubuntu16.04": "sha256:b90dcf2f35f3354909f4491bdf019c110b4b4d95ef0395ebf178bc5d523a4208",
15    "centos6": "sha256:d09c12fb26fbbe8398b4973260c75172eb67d509dae9d6f4ad54279b7d6b0494",
16    "ubuntu16.04-manylinux2010": "sha256:5b6a55a33046843a3c3b3b3d13b028cc6e61d8af5275998e95fa539cafe72c51",
17    "cuda10.0-cudnn7-ubuntu14.04": "sha256:d433e1221f802dac393bc8652fabcc63aa46896cd920bb888ae0e2002fe6b756",
18    "cuda10.0-cudnn7-centos7": "sha256:a453b7147a60928a8345689eae48916a746b3578b5e831bfa151f0529d469c88",
19    "cuda10.0-cudnn7-centos6": "sha256:a1909ba09c703340ee0074ce63dd94fe8fea48035a25264677907a609e2375e0",
20    "cuda10.1-cudnn7-centos6": "sha256:454b899657e87893ee5e68dc0f87df59b6a0a7418ae09cafcc3dd65ac71feca9",
21    "cuda10.0-cudnn7-ubuntu16.04-manylinux2010": "sha256:5812d9d0ef0a3276fc5faaf4cd01f3d6e03d635893a6e2d2e04f6f01d626c432",
22    "cuda10.1-cudnn7-ubuntu16.04-manylinux2010": "sha256:5e6d21c8ef226316eb6df5e2e6015244c16a8e5d936b52a09820442d2f8a919f",
23    "cuda10.1-cudnn7-ubuntu16.04-manylinux2010-multipython": "sha256:3f890a951c81a201d60d0161a56ce628a90323be0c7f795550caa37f6f41a85c",
24    "cuda10.1-cudnn7-ubuntu18.04-manylinux2010-multipython": "sha256:71e946b59a63e80d10178f9e2869af3b31583737b779edc27688c99f4998dd05",
25    "cuda10.2-cudnn7-ubuntu18.04-manylinux2010-multipython": "sha256:9bb24f6f98e71b424dd885ec9fe39770831376840fa2231057807c89f814c803",
26    "cuda11.0-cudnn8-ubuntu18.04-manylinux2010-multipython": "sha256:3e5c991f67e2cca610cb9f6b39927e3757ba1e7f2424d18cef8b871bfa4d75b3",
27    "windows-1803": "sha256:f109576c7c0c8a1783ff22b666e8923b52dbbe7933f69a1c7a7275202c304a12",
28}
29
30containers = {
31    # Built with //tensorflow/tools/ci_build/Dockerfile.rbe.ubuntu16.04-manylinux2010.
32    "ubuntu16.04-manylinux2010": {
33        "registry": "gcr.io",
34        "repository": "tensorflow-testing/nosla-ubuntu16.04-manylinux2010",
35        "digest": container_digests["ubuntu16.04-manylinux2010"],
36    },
37
38    # Built with //tensorflow/tools/ci_build/Dockerfile.rbe.cuda10.0-cudnn7-ubuntu16.04-manylinux2010.
39    "cuda10.0-cudnn7-ubuntu16.04-manylinux2010": {
40        "registry": "gcr.io",
41        "repository": "tensorflow-testing/nosla-cuda10.0-cudnn7-ubuntu16.04-manylinux2010",
42        "digest": container_digests["cuda10.0-cudnn7-ubuntu16.04-manylinux2010"],
43    },
44
45    # Built with //tensorflow/tools/ci_build/Dockerfile.rbe.cuda10.1-cudnn7-ubuntu16.04-manylinux2010.
46    "cuda10.1-cudnn7-ubuntu16.04-manylinux2010": {
47        "registry": "gcr.io",
48        "repository": "tensorflow-testing/nosla-cuda10.1-cudnn7-ubuntu16.04-manylinux2010",
49        "digest": container_digests["cuda10.1-cudnn7-ubuntu16.04-manylinux2010"],
50    },
51
52    # Built with //tensorflow/tools/ci_build/Dockerfile.rbe.cuda10.1-cudnn7-ubuntu16.04-manylinux2010-multipython.
53    "cuda10.1-cudnn7-ubuntu16.04-manylinux2010-multipython": {
54        "registry": "gcr.io",
55        "repository": "tensorflow-testing/nosla-cuda10.1-cudnn7-ubuntu16.04-manylinux2010-multipython",
56        "digest": container_digests["cuda10.1-cudnn7-ubuntu16.04-manylinux2010-multipython"],
57    },
58
59    # Built with //tensorflow/tools/ci_build/Dockerfile.rbe.cuda10.1-cudnn7-ubuntu18.04-manylinux2010-multipython.
60    "cuda10.1-cudnn7-ubuntu18.04-manylinux2010-multipython": {
61        "registry": "gcr.io",
62        "repository": "tensorflow-testing/nosla-cuda10.1-cudnn7-ubuntu18.04-manylinux2010-multipython",
63        "digest": container_digests["cuda10.1-cudnn7-ubuntu18.04-manylinux2010-multipython"],
64    },
65
66    # Built with //tensorflow/tools/ci_build/Dockerfile.rbe.cuda10.2-cudnn7-ubuntu18.04-manylinux2010-multipython.
67    "cuda10.2-cudnn7-ubuntu18.04-manylinux2010-multipython": {
68        "registry": "gcr.io",
69        "repository": "tensorflow-testing/nosla-cuda10.2-cudnn7-ubuntu18.04-manylinux2010-multipython",
70        "digest": container_digests["cuda10.2-cudnn7-ubuntu18.04-manylinux2010-multipython"],
71    },
72
73    # Built with //tensorflow/tools/ci_build/Dockerfile.rbe.cuda11.0-cudnn8-ubuntu18.04-manylinux2010-multipython.
74    "cuda11.0-cudnn8-ubuntu18.04-manylinux2010-multipython": {
75        "registry": "gcr.io",
76        "repository": "tensorflow-testing/nosla-cuda11.0-cudnn8-ubuntu18.04-manylinux2010-multipython",
77        "digest": container_digests["cuda11.0-cudnn8-ubuntu18.04-manylinux2010-multipython"],
78    },
79
80    # Built with //tensorflow/tools/ci_build/Dockerfile.rbe.cuda11.2-cudnn8.1-ubuntu20.04-manylinux2014-multipython.
81    "cuda11.2-cudnn8.1-ubuntu20.04-manylinux2014-multipython": {
82        "registry": "gcr.io",
83        "repository": "tensorflow-testing/nosla-cuda11.2-cudnn8.1-ubuntu20.04-manylinux2014-multipython",
84        "digest": container_digests["cuda11.2-cudnn8.1-ubuntu20.04-manylinux2014-multipython"],
85    },
86
87    # Built with //tensorflow/tools/ci_build/Dockerfile.rbe.cuda11.1-cudnn8-ubuntu20.04-manylinux2014-multipython.
88    "cuda11.1-cudnn8-ubuntu20.04-manylinux2014-multipython": {
89        "registry": "gcr.io",
90        "repository": "tensorflow-testing/nosla-cuda11.1-cudnn8-ubuntu20.04-manylinux2014-multipython",
91        "digest": container_digests["cuda11.1-cudnn8-ubuntu20.04-manylinux2014-multipython"],
92    },
93
94    # Built with //tensorflow/tools/ci_build/Dockerfile.rbe.cuda11.4-cudnn8.2-ubuntu20.04-manylinux2014-multipython.
95    "cuda11.4-cudnn8.2-ubuntu20.04-manylinux2014-multipython": {
96        "registry": "gcr.io",
97        "repository": "tensorflow-testing/nosla-cuda11.4-cudnn8.2-ubuntu20.04-manylinux2014-multipython",
98        "digest": container_digests["cuda11.4-cudnn8.2-ubuntu20.04-manylinux2014-multipython"],
99    },
100
101    # Built with //tensorflow/tools/ci_build/Dockerfile.rbe.rocm-ubuntu18.04-manylinux2010-multipython.
102    "rocm-ubuntu18.04-manylinux2010-multipython": {
103        "registry": "gcr.io",
104        "repository": "tensorflow-testing/nosla-rocm-ubuntu18.04-manylinux2010-multipython",
105        "digest": container_digests["rocm-ubuntu18.04-manylinux2010-multipython"],
106    },
107
108    # Built with //tensorflow/tools/ci_build/Dockerfile.rbe.rocm-ubuntu20.04-manylinux2014-multipython.
109    "rocm-ubuntu20.04-manylinux2014-multipython": {
110        "registry": "gcr.io",
111        "repository": "tensorflow-testing/nosla-rocm-ubuntu20.04-manylinux2014-multipython",
112        "digest": container_digests["rocm-ubuntu20.04-manylinux2014-multipython"],
113    },
114
115    # Built by mihaimaruseac@ from a private Dockerfile.
116    "windows-1803": {
117        "registry": "gcr.io",
118        "repository": "tensorflow-testing/tf-win-rbe",
119        "digest": container_digests["windows-1803"],
120    },
121}
122