Commit 93d7198d authored by Jeremy Auguste's avatar Jeremy Auguste

Changed it so environment names are less restrictive

parent f167d828
......@@ -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
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment