From 08739cdd0a988d5bef1ddfc1d817d347c7c09256 Mon Sep 17 00:00:00 2001 From: oeway Date: Mon, 28 Sep 2020 23:23:59 +0200 Subject: [PATCH] Fix error in render_tensorflow --- pydiffvg_tensorflow/render_tensorflow.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pydiffvg_tensorflow/render_tensorflow.py b/pydiffvg_tensorflow/render_tensorflow.py index 6a7875b..eed7e28 100644 --- a/pydiffvg_tensorflow/render_tensorflow.py +++ b/pydiffvg_tensorflow/render_tensorflow.py @@ -1,3 +1,4 @@ +import os import tensorflow as tf import diffvg import pydiffvg_tensorflow as pydiffvg @@ -457,7 +458,7 @@ def render(*x): The main TensorFlow interface of C++ diffvg. """ assert(tf.executing_eagerly()) - if pydiffvg.get_use_gpu() and os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] != 'true': + if pydiffvg.get_use_gpu() and os.environ.get('TF_FORCE_GPU_ALLOW_GROWTH') != 'true': print('******************** WARNING ********************') print('Tensorflow by default allocates all GPU memory,') print('causing huge amount of page faults when rendering.')