From b302ae215aec20fdd4718a99ee0ca4874fcbe1be Mon Sep 17 00:00:00 2001 From: Andrei Nechaev Date: Fri, 4 May 2018 11:01:45 -0400 Subject: [PATCH] path to nvcc --- nvcc_plugin.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/nvcc_plugin.py b/nvcc_plugin.py index 7f531c9..c6dc1b0 100644 --- a/nvcc_plugin.py +++ b/nvcc_plugin.py @@ -6,6 +6,7 @@ import tempfile import subprocess import IPython.core.magic as ipym +compiler = '/usr/local/cuda/bin/nvcc' ext = '.cu' @@ -25,7 +26,7 @@ class NVCCPlugin(ipym.Magics): @staticmethod def compile(file_path): - subprocess.check_output(["nvcc", file_path + ext, "-o", file_path + ".out"], stderr=subprocess.STDOUT) + subprocess.check_output([compiler, file_path + ext, "-o", file_path + ".out"], stderr=subprocess.STDOUT) def run(self, file_path, timeit=False): if timeit: