Wrap NervanaGPU import into a try/catch block

Running on CPU-only machines invokes an error when this code attempts to import NervanaGPU.
NervanaSystems · Jul 7, 2016 · a01a2d9 · a01a2d9
1 parent b8b8b15
commit a01a2d9
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/neon/backends/__init__.py b/neon/backends/__init__.py
@@ -28,7 +28,11 @@
 
 # These are imported to register the backends with the factory
 import neon.backends.nervanacpu
-import neon.backends.nervanagpu
+
+try:
+    import neon.backends.nervanagpu
+except ImportError:
+    pass
 
 try:
     # Register if it exists