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

Added the pytorch environment

parent 817f3d65
No related branches found
No related tags found
No related merge requests found
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment