commit 4c51a523c8401588b6fbdef5a3d5870b68099fd6
author Adam Paszke <adam.paszke@gmail.com> Fri Aug 19 14:20:44 2016 -0700
committer Adam Paszke <adam.paszke@gmail.com> Fri Aug 19 14:23:53 2016 -0700
tree f394663d67ca29716ce0e1eeb5d8bc606a14af2a
parent aadac369dc82ae9667d93f96ec6ac1af1c223921

Add super basic CUDA autodetection

setup.py

diff --git a/setup.py b/setup.py
index d80f2d1..a3597e1 100644
--- a/setup.py
+++ b/setup.py
@@ -9,8 +9,8 @@
 import sys
 import os
 
-# TODO: detect CUDA
-WITH_CUDA = False
+# TODO: make this more robust
+WITH_CUDA = os.path.exists('/Developer/NVIDIA/CUDA-7.5/include') or os.path.exists('/usr/local/cuda/include')
 DEBUG = False
 
 ################################################################################