make torch.cuda.empty_cache() a no-op when cuda is not initialized (#4936)

soumith · soumith · commit 00f9da74b3bb · 2018-02-07T09:15:29.000-08:00
diff --git a/torch/cuda/__init__.py b/torch/cuda/__init__.py
@@ -322,8 +322,8 @@ def empty_cache():
     """Releases all unoccupied cached memory currently held by the caching
     allocator so that those can be used in other GPU application and visible in
     `nvidia-smi`."""
-    _lazy_init()
-    return torch._C._cuda_emptyCache()
+    if _initialized:
+        return torch._C._cuda_emptyCache()
 
 
 def _host_allocator():
