TFHUB_CACHE_DIR

1 files changed, 5 insertions, 1 deletions

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,
   }