forked from Superalgos/Superalgos
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtfSetup.js
More file actions
27 lines (22 loc) · 815 Bytes
/
tfSetup.js
File metadata and controls
27 lines (22 loc) · 815 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
const fs = require('fs')
const os = require('os')
const path = require('path')
const tfSetup = () => {
const srcFolder = './node_modules/@tensorflow/tfjs-node/deps/lib/'
const destFolder = './node_modules/@tensorflow/tfjs-node/lib/'
if (os.platform() === 'win32' && fs.existsSync(srcFolder) && fs.existsSync(destFolder)) {
const dest = fs.readdirSync(destFolder, { withFileTypes: true })
.filter(dirent => dirent.isDirectory())
.map(dirent => dirent.name)
fs.copyFile(path.join(srcFolder, "tensorflow.dll"), path.join(destFolder, dest[0], "tensorflow.dll"), (err) => {
if (err) {
console.log(err)
return err
}
console.log('tensorflow dll copied')
})
return 'tensorflow dll copied'
}
return 'non-windows system'
}
module.exports = tfSetup