From 4b8386c3ba5fcb5eee430ad205e96f429e44f45a Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Fri, 20 Sep 2019 22:34:54 +0200
Subject: [PATCH] colab
---
src/pystencils_autodiff/tensorflow_jit.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/pystencils_autodiff/tensorflow_jit.py b/src/pystencils_autodiff/tensorflow_jit.py
index 3b90107..e5312b0 100644
--- a/src/pystencils_autodiff/tensorflow_jit.py
+++ b/src/pystencils_autodiff/tensorflow_jit.py
@@ -24,7 +24,7 @@ _hash = hashlib.md5
if 'NVCC_BINARY' in os.environ:
NVCC_BINARY = os.environ['NVCC_BINARY']
else:
- NVCC_BINARY = 'nvcc'
+ NVCC_BINARY = '/usr/local/cuda/bin/nvcc'
# TODO: msvc
if get_compiler_config()['os'] != 'windows':
--
GitLab