Update Jenkinsfile

xingzuhui · Jun 17, 2021 · df95008 · df95008
1 parent 7cc2c24
commit df95008
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/Jenkinsfile b/Jenkinsfile
@@ -46,6 +46,7 @@ stage("Build and Publish") {
       conda activate ${ENV_NAME}
       ./static/cache.sh restore _build/eval_tensorflow/data
       export TF_CPP_MIN_LOG_LEVEL=3
+      export TF_FORCE_GPU_ALLOW_GROWTH=true
       d2lbook build eval --tab tensorflow
       ./static/cache.sh store _build/eval_tensorflow/data
       """