Change GPU check to CUDA SDK check. Closes issue #19

NervanaSystems · May 8, 2015 · e519b81 · e519b81
1 parent 6c9e644
commit e519b81
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Makefile b/Makefile
@@ -46,7 +46,7 @@ ifneq ($(GPU), 0)
     endif
   else
     # we assume a Linux-like OS
-    ifneq ($(shell nvidia-smi > /dev/null 2>&1; echo $$?), 0)
+    ifneq ($(shell nvcc --version > /dev/null 2>&1; echo $$?), 0)
       $(info No CUDA capable GPU installed.  Forcing GPU=0)
       override GPU := 0
     endif