diff --git a/pylauncher.sh b/pylauncher.sh
index a7318d26a87d09f7a8a66dff5b3bbd104d3b7e86..ae96ee3149925c99b1d752c065cce445bcb09e8e 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 "$@"