refine the code test=develop

PaddlePaddle · Aug 16, 2020 · 89c8635 · 89c8635 · paddle-bot-old · Aug 16, 2020
1 parent aa24075
commit 89c8635
Showing 1 changed file with 6 additions and 3 deletions.
diff --git 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