Add super basic CUDA autodetection

2025-10-21 13:44:15 +08:00 · 2016-08-19 14:20:44 -07:00
parent aadac369dc
commit 4c51a523c8
1 changed files with 2 additions and 2 deletions
--- a/setup.py
+++ b/setup.py
@ -9,8 +9,8 @@ import shutil
 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

 ################################################################################