Skip to content
Permalink
Browse files

minor change in jenkins script, can install tensorflow CPU/GPU with b…

…ash input argument
  • Loading branch information...
chanlevan committed Apr 16, 2019
1 parent 71bc0cd commit a4289f22c887e6fe84dbcd4ad021a30b23aad307
Showing with 9 additions and 1 deletion.
  1. +9 −1 jenkins.sh
@@ -9,7 +9,15 @@ conda create --name ampligraph python=3.6
source activate ampligraph

# Install library
pip install tensorflow==1.12.0
if [[ $# -eq 0 ]] ; then
echo "install tensorflow CPU mode"
pip install tensorflow==1.12.0
else
if [[ $1 == "gpu" ]] ; then
echo "install tensorflow GPU mode"
conda install tensorflow-gpu==1.12.0
fi
fi

pip install . -v

0 comments on commit a4289f2

Please sign in to comment.
You can’t perform that action at this time.