From d207237a285fa27733188558345f76dbc6b3f74f Mon Sep 17 00:00:00 2001 From: Leo Fang Date: Sun, 26 May 2019 23:47:40 -0400 Subject: [PATCH] updated run-ptycho to support Numba in Linux --- run-ptycho | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/run-ptycho b/run-ptycho index d3c0813..7ad5ec8 100755 --- a/run-ptycho +++ b/run-ptycho @@ -7,6 +7,10 @@ export LD_LIBRARY_PATH=${CUDA_HOME}/lib64:$LD_LIBRARY_PATH export LD_LIBRARY_PATH=${CUDA_HOME}/extras/CUPTI/lib64:$LD_LIBRARY_PATH export LD_LIBRARY_PATH=${CUDA_HOME}/nvvm/lib64:$LD_LIBRARY_PATH +# Numba +export NUMBAPRO_LIBDEVICE=/usr/local/cuda/nvvm/libdevice/ +export NUMBAPRO_NVVM=/usr/local/cuda/nvvm/lib64/libnvvm.so + # use the production conda environment source /opt/conda/bin/activate ptycho_production