diff options
| -rw-r--r-- | app/relay/modules/biggan.js | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/app/relay/modules/biggan.js b/app/relay/modules/biggan.js index 30e28a0..c07a04b 100644 --- a/app/relay/modules/biggan.js +++ b/app/relay/modules/biggan.js @@ -4,8 +4,12 @@ import fs from 'fs' const name = 'biggan' const cwd = process.env.BIGGAN_CWD || path.join(process.env.HOME, 'code/' + name + '/cli/') const env = { - LD_LIBRARY_PATH: '/usr/local/cuda/lib64:/home/lens/ffmpeg_build/lib:' + process.env.TENSORRT_LIB_PATH, - TFHUB_CACHE_DIR: '/home/lens/.tfhub', + LD_LIBRARY_PATH: [ + '/usr/local/cuda/lib64', + process.env.FFMPEG_LIB_PATH, + process.env.TENSORRT_LIB_PATH, + ].filter(s => !!s).join(':'), + TFHUB_CACHE_DIR: process.env.HOME + '/.tfhub', LC_ALL: 'C.UTF-8', LANG: 'C.UTF-8', } |