From 6d46dc057f7a557e91c98d96eee1b6aa84119208 Mon Sep 17 00:00:00 2001
From: Jeremy Auguste <jeremy.auguste@etu.univ-amu.fr>
Date: Thu, 6 Jul 2017 11:09:28 +0200
Subject: [PATCH] Temporary fix for CUDA in keras environment

---
 pylauncher.sh | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/pylauncher.sh b/pylauncher.sh
index a7318d2..ae96ee3 100755
--- a/pylauncher.sh
+++ b/pylauncher.sh
@@ -11,4 +11,12 @@ shift 1
 
 source activate $environment
 
+if [[ $environment == "keras" ]]; then
+    export CUDA_HOME=/usr/local/cuda-8.0
+    export CUDA_ROOT=$CUDA_HOME
+	export PATH=$CUDA_HOME/bin:$PATH
+	export MANPATH=$CUDA_HOME/doc/man:$MANPATH
+	export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH
+fi
+
 eval "$@"
-- 
GitLab