ShinobiSystems
/
Shinobi
mirror of https://gitlab.com/Shinobi-Systems/Shinobi.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

update tensorflow plugin install post action dir

fix-non-showing-inputs
Moe Alam 2020-11-19 16:16:50 -08:00
parent f2a84dfede
commit 5a79a55e4b
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
plugins/tensorflow/INSTALL.sh
View File

@ -102,12 +102,12 @@ else
installJetson
armAfterInstall
fi
if [ "$installArmFlag" = true ]; then
installArm
armAfterInstall
fi
if [ "$installGpuFlag" = true ]; then
installGpuRoute
else
@ -131,9 +131,9 @@ if [ "$dontCreateKeyFlag" = false ]; then
if [ "$installGpuFlag" = true ]; then
tfjsBuildVal="gpu"
fi
echo "Adding Random Plugin Key to Main Configuration"
node /tools/modifyConfigurationForPlugin.js tensorflow key=$(head -c 64 < /dev/urandom | sha256sum | awk '{print substr($1,1,60)}') tfjsBuild=$tfjsBuildVal
node $DIR/../../tools/modifyConfigurationForPlugin.js tensorflow key=$(head -c 64 < /dev/urandom | sha256sum | awk '{print substr($1,1,60)}') tfjsBuild=$tfjsBuildVal
fi
echo "TF_FORCE_GPU_ALLOW_GROWTH=true" > "$DIR/.env"