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

Update plugin TensorFlow GPU Install

build-default-monitor-config-from-definitions
Moe 2020-04-26 11:57:47 -07:00
parent 5a63dd7385
commit 474a3c599f
3 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
plugins/tensorflow/.env Normal file
View File

@ -0,0 +1,2 @@
TF_FORCE_GPU_ALLOW_GROWTH=true
#CUDA_VISIBLE_DEVICES=0,2

5
plugins/tensorflow/INSTALL.sh
View File

@ -5,6 +5,7 @@ echo "You can run this installer again to change it."
echo "(y)es or (N)o" echo "(y)es or (N)o"
read nodejsinstall read nodejsinstall
echo "Getting Tensorflow Node.js module..." echo "Getting Tensorflow Node.js module..."
npm install dotenv@8.2.0 --unsafe-perm
npm uninstall @tensorflow/tfjs-node-gpu --unsafe-perm npm uninstall @tensorflow/tfjs-node-gpu --unsafe-perm
npm uninstall @tensorflow/tfjs-node --unsafe-perm npm uninstall @tensorflow/tfjs-node --unsafe-perm
npm install @tensorflow/tfjs-core@1.7.3 --unsafe-perm --force npm install @tensorflow/tfjs-core@1.7.3 --unsafe-perm --force
@ -12,7 +13,9 @@ npm install @tensorflow/tfjs-converter@1.7.3 --unsafe-perm --force
npm install @tensorflow/tfjs-layers@1.7.3 --unsafe-perm --force npm install @tensorflow/tfjs-layers@1.7.3 --unsafe-perm --force
npm install yarn -g --unsafe-perm --force npm install yarn -g --unsafe-perm --force
npm install @tensorflow/tfjs-node@1.7.3 --unsafe-perm npm install @tensorflow/tfjs-node@1.7.3 --unsafe-perm
GPU_INSTALL="0"
if [ "$nodejsinstall" = "y" ] || [ "$nodejsinstall" = "Y" ]; then if [ "$nodejsinstall" = "y" ] || [ "$nodejsinstall" = "Y" ]; then
GPU_INSTALL="1"
npm install @tensorflow/tfjs-node-gpu@1.7.0 --unsafe-perm npm install @tensorflow/tfjs-node-gpu@1.7.0 --unsafe-perm
fi fi
echo "Getting Coco SSD Model..." echo "Getting Coco SSD Model..."
@ -26,7 +29,7 @@ else
fi fi
echo "Adding Random Plugin Key to Main Configuration" echo "Adding Random Plugin Key to Main Configuration"
tfjsBuildVal="cpu" tfjsBuildVal="cpu"
if [ "$nodejsinstall" = "1" ]; then if [ "$GPU_INSTALL" = "1" ]; then
tfjsBuildVal="gpu" tfjsBuildVal="gpu"
fi fi
node $DIR/../../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

1
plugins/tensorflow/shinobi-tensorflow.js
View File

@ -10,6 +10,7 @@
// Base Init >> // Base Init >>
var fs = require('fs'); var fs = require('fs');
var config = require('./conf.json') var config = require('./conf.json')
var dotenv = require('dotenv').config()
var s var s
try{ try{
s = require('../pluginBase.js')(__dirname,config) s = require('../pluginBase.js')(__dirname,config)