read python executable

build.py

+import sys
 import os
 import shutil
 import subprocess
@@ -16,7 +17,7 @@ extra_objects = []
 with_cuda = False
 
 if torch.cuda.is_available():
-    subprocess.call('./build.sh')
+    subprocess.call('./build.sh {}'.format(sys.executable))
 
     headers += ['torch_spline_conv/src/cuda.h']
     sources += ['torch_spline_conv/src/cuda.c']

build.sh

@@ -2,7 +2,8 @@
 
 echo "Compiling kernel..."
 
-TORCH=$(python -c "import os; import torch; print(os.path.dirname(torch.__file__))")
+PYTHON=${1:-python}
+TORCH=$($PYTHON -c "import os; import torch; print(os.path.dirname(torch.__file__))")
 SRC_DIR=torch_spline_conv/kernel
 BUILD_DIR=torch_spline_conv/build