Add torch.cuda.cudnn_is_available (#8703)

goldsborough · web-flow · commit 0acddd6cee32 · 2018-06-20T14:18:03.000-07:00
diff --git a/torch/backends/cudnn/__init__.py b/torch/backends/cudnn/__init__.py
@@ -74,12 +74,17 @@ def version():
 }
 
 
+def is_available():
+    r"""Returns a bool indicating if CUDNN is currently available."""
+    return torch._C.has_cudnn
+
+
 def is_acceptable(tensor):
     if not torch._C._get_cudnn_enabled():
         return False
     if tensor.type() not in CUDNN_TENSOR_TYPES:
         return False
-    if not torch._C.has_cudnn:
+    if not is_available():
         warnings.warn(
             "PyTorch was compiled without cuDNN support. To use cuDNN, rebuild "
             "PyTorch making sure the library is visible to the build system.")
