From 5a79a55e4bb7b58ecdcf9217462a6503ec02d41b Mon Sep 17 00:00:00 2001 From: Moe Alam Date: Thu, 19 Nov 2020 16:16:50 -0800 Subject: [PATCH] update tensorflow plugin install post action dir --- plugins/tensorflow/INSTALL.sh | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/plugins/tensorflow/INSTALL.sh b/plugins/tensorflow/INSTALL.sh index 8039ac58..a16d7c1f 100644 --- a/plugins/tensorflow/INSTALL.sh +++ b/plugins/tensorflow/INSTALL.sh @@ -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"