From 93d7198db5961dd9531d2f3308c0c723cd61d410 Mon Sep 17 00:00:00 2001
From: Jeremy Auguste <jeremy.auguste@dhcp-154.lidil.univ-mrs.fr>
Date: Mon, 22 Jan 2018 09:49:04 +0100
Subject: [PATCH] Changed it so environment names are less restrictive
---
pylauncher.sh | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/pylauncher.sh b/pylauncher.sh
index 21430bb..a81720a 100755
--- a/pylauncher.sh
+++ b/pylauncher.sh
@@ -10,7 +10,7 @@ shift 1
source activate $environment
-if [[ $environment == "keras" || $environment == "pytorch" ]]; then
+if [[ $environment =~ ^keras.* || $environment =~ ^pytorch.* ]]; then
export CUDA_HOME=/usr/local/cuda-8.0
export CUDA_ROOT=$CUDA_HOME
export PATH=$CUDA_HOME/bin:$PATH
--
GitLab