From a0fa03a9a0219771e0173db5a811d20b20fe3ba0 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Thu, 19 Dec 2019 17:11:39 +0100 Subject: TFHUB_CACHE_DIR --- app/relay/modules/biggan.js | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/app/relay/modules/biggan.js b/app/relay/modules/biggan.js index 97042be..9dfa6f7 100644 --- a/app/relay/modules/biggan.js +++ b/app/relay/modules/biggan.js @@ -3,6 +3,10 @@ import fs from 'fs' const name = 'biggan' const cwd = process.env.BIGGAN_CWD || path.join(process.env.HOME, 'code/' + name + '/') +const env = { + LD_LIBRARY_PATH: '/usr/local/cuda/lib64:' + process.env.TENSORRT_LIB_PATH, + TFHUB_CACHE_DIR: '/home/lens/.tfhub', +} const live = { type: 'pytorch_biggan', script: 'inversion/live.py', @@ -14,7 +18,7 @@ const live = { } export default { - name, cwd, + name, cwd, env, activities: { live, } -- cgit v1.2.3-70-g09d2