diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2020-01-17 11:22:59 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2020-01-17 11:22:59 +0100 |
| commit | 0e97d25764dac9c9dc8ce38835b13f6875010575 (patch) | |
| tree | 24273034971aec311cfac3706b983789b7230f08 /app | |
| parent | 1ea4a497043ca3f4efe246d6488e3ce57574bf3b (diff) | |
paths
Diffstat (limited to 'app')
| -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', } |