Skip to content
Snippets Groups Projects
Commit 93d7198d authored by Jeremy Auguste's avatar Jeremy Auguste
Browse files

Changed it so environment names are less restrictive

parent f167d828
No related branches found
No related tags found
No related merge requests found
...@@ -10,7 +10,7 @@ shift 1 ...@@ -10,7 +10,7 @@ shift 1
source activate $environment 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_HOME=/usr/local/cuda-8.0
export CUDA_ROOT=$CUDA_HOME export CUDA_ROOT=$CUDA_HOME
export PATH=$CUDA_HOME/bin:$PATH export PATH=$CUDA_HOME/bin:$PATH
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment