From a446b75333ebf0216a0bfc6e345fdee553a7e1eb Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Mon, 18 Jun 2018 15:57:54 +0200 Subject: img opz --- app/relay/modules/pix2pixhd.js | 7 +++++-- app/relay/runner.js | 1 + 2 files changed, 6 insertions(+), 2 deletions(-) (limited to 'app') diff --git a/app/relay/modules/pix2pixhd.js b/app/relay/modules/pix2pixhd.js index 5b9fd5c..34fc972 100644 --- a/app/relay/modules/pix2pixhd.js +++ b/app/relay/modules/pix2pixhd.js @@ -3,6 +3,10 @@ import fs from 'fs' const name = 'pix2pixhd' const cwd = process.env.PIX2PIXHD_CWD || path.join(process.env.HOME, 'code/' + name + '/') +const env = { + LD_LIBRARY_PATH: process.env.HOME + '/Downloads/TensorRT-3.0.4/lib', +} +console.log(env) const fetch = { type: 'perl', @@ -101,7 +105,6 @@ const live = { console.log(task) const opt = task.opt || {} return [ - 'LD_LIBRARY_PATH=' + process.env.HOME + '/Downloads/TensorRT-3.0.4/lib', '--dataroot', path.join(cwd, 'sequences', task.dataset), '--start_img', path.join(cwd, 'sequences', task.dataset, 'frame_00001.png'), '--checkpoint-name', task.checkpoint, @@ -128,7 +131,7 @@ const live = { } export default { - name, cwd, + name, cwd, env, activities: { fetch, build, diff --git a/app/relay/runner.js b/app/relay/runner.js index 6464288..531bf12 100644 --- a/app/relay/runner.js +++ b/app/relay/runner.js @@ -261,6 +261,7 @@ export function run_task_with_activity(task, module, activity, preempt=false, wa task.processing = true const subprocess = spawn(interpreter.cmd, params, { cwd: module.cwd, + env: module.env || {}, }) if (interpreter.gpu) { state.current_gpu_task = { -- cgit v1.2.3-70-g09d2