From 89c863504efbc58958f6fd888bbfe72ad1ec5a25 Mon Sep 17 00:00:00 2001 From: wangchaochaohu Date: Sun, 16 Aug 2020 01:34:01 +0000 Subject: [PATCH] refine the code test=develop --- python/paddle/device.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/python/paddle/device.py b/python/paddle/device.py index bda9440571acc..e2ef8e7092ad3 100644 --- a/python/paddle/device.py +++ b/python/paddle/device.py @@ -52,9 +52,12 @@ def get_cudnn_version(): """ global _cudnn_version - if _cudnn_version is None and core.is_compiled_with_cuda(): - cudnn_version = core.cudnn_version() - if cudnn_version < 0: + if not core.is_compiled_with_cuda(): + return None + if _cudnn_version is None: + cudnn_version = int(core.cudnn_version()) + _cudnn_version = cudnn_version + if _cudnn_version < 0: return None else: return cudnn_version