From a54cee34245f2f14f676adeefa1d86f4968d62ca Mon Sep 17 00:00:00 2001
From: Jeremy Auguste <jeremy.auguste@dhcp-13.lidil.univ-mrs.fr>
Date: Tue, 6 Mar 2018 11:52:56 +0100
Subject: [PATCH] Added options and more flexibility with cuda
---
pylauncher.sh | 31 +++++++++++++++++++++++++++++--
1 file changed, 29 insertions(+), 2 deletions(-)
diff --git a/pylauncher.sh b/pylauncher.sh
index a81720a..cbae654 100755
--- a/pylauncher.sh
+++ b/pylauncher.sh
@@ -1,17 +1,44 @@
#! /bin/zsh
if [[ $# -lt 2 ]]; then
- echo "Usage: $0 environment_name command [arg1..argn]" >&2
+ echo "Usage: $0 environment_name [OPTIONS] command [arg1..argn]" >&2
+ echo "OPTIONS:" >&2
+ echo " -h, --help Show this message" >&2
+ echo " -p, --cuda-path Specify the path where cuda is installed (default: /usr/local/cuda-8.0)" >&2
exit 1
fi
+cuda_path="/usr/local/cuda-8.0"
+
+while [[ $1 == -* ]]; do
+ case "$1" in
+ -p|--cuda-path)
+ cuda_path="$2"
+ shift 2
+ ;;
+ -h|--help)
+ echo "Usage: $0 environment_name [OPTIONS] command [arg1..argn]"
+ echo "OPTIONS:"
+ echo " -h, --help Show this message"
+ echo " -p, --cuda-path Specify the path where cuda is installed (default: /usr/local/cuda-8.0)"
+ exit 0
+ ;;
+ *)
+ echo "Error: Unknown option: $1" >&2
+ echo "Usage: $0 environment_name [OPTIONS] command [arg1..argn]" >&2
+ exit 1
+ ;;
+ esac
+done
+
+
environment="$1"
shift 1
source activate $environment
if [[ $environment =~ ^keras.* || $environment =~ ^pytorch.* ]]; then
- export CUDA_HOME=/usr/local/cuda-8.0
+ export CUDA_HOME=$cuda_path
export CUDA_ROOT=$CUDA_HOME
export PATH=$CUDA_HOME/bin:$PATH
export MANPATH=$CUDA_HOME/doc/man:$MANPATH
--
GitLab