()
| 15 | |
| 16 | |
| 17 | def parse_TF_build_info(): |
| 18 | ret = OrderedDict() |
| 19 | from tensorflow.python.platform import build_info |
| 20 | try: |
| 21 | for k, v in list(build_info.build_info.items()): |
| 22 | if k == "cuda_version": |
| 23 | ret["TF built with CUDA"] = v |
| 24 | elif k == "cudnn_version": |
| 25 | ret["TF built with CUDNN"] = v |
| 26 | elif k == "cuda_compute_capabilities": |
| 27 | ret["TF compute capabilities"] = ",".join([k.replace("compute_", "") for k in v]) |
| 28 | return ret |
| 29 | except AttributeError: |
| 30 | pass |
| 31 | try: |
| 32 | ret["TF built with CUDA"] = build_info.cuda_version_number |
| 33 | ret["TF built with CUDNN"] = build_info.cudnn_version_number |
| 34 | except AttributeError: |
| 35 | pass |
| 36 | return ret |
| 37 | |
| 38 | |
| 39 | def collect_env_info(): |
no test coverage detected
searching dependent graphs…