Temporary fix for CUDA in keras environment

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 "$@"