From 24fc0d4b03aa99023ed438b66a6b213cdf356c33 Mon Sep 17 00:00:00 2001
From: Jeremy Auguste <jeremy.auguste@etu.univ-amu.fr>
Date: Thu, 9 Nov 2017 09:54:00 +0100
Subject: [PATCH] Added the pytorch environment

---
 pylauncher.sh | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/pylauncher.sh b/pylauncher.sh
index ae96ee3..21430bb 100755
--- a/pylauncher.sh
+++ b/pylauncher.sh
@@ -7,11 +7,10 @@ fi
 
 environment="$1"
 shift 1
-# command="$2"
 
 source activate $environment
 
-if [[ $environment == "keras" ]]; 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