caffenode-js-build
Version:
build caffe with GPU/CPU
48 lines (39 loc) • 639 B
JavaScript
const {
opencvBuild,
opencvInclude,
opencvLibDir,
cudaInclude,
cudaLib,
cudaLib64,
caffeBuild,
caffeProtoDir,
caffeSrc,
caffeInclude,
caffeLibDir
} = require('./config');
const {
isCPU,
hasGPU
} = require('./device');
const {
getLibs
} = require('./libs');
module.exports = {
// build
caffeBuild,
opencvBuild,
// includes
caffeInclude,
caffeProtoDir,
opencvInclude,
cudaInclude,
// libs
opencvLibDir,
caffeLibDir,
cudaLib,
cudaLib64,
// fetching required libs
libs: getLibs(),
// GPU / CPU
isCPU: isCPU()
}