Skip to content

S
seminar-breakout
Commit 1097672b authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

fix #1074

parent d24a9230
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
tensorpack/libinfo.py
View file @ 1097672b
...@@ -45,8 +45,9 @@ os.environ['TF_GPU_THREAD_COUNT'] = '2' ...@@ -45,8 +45,9 @@ os.environ['TF_GPU_THREAD_COUNT'] = '2'
os.environ['TF_USE_CUDNN_BATCHNORM_SPATIAL_PERSISTENT'] = '0' os.environ['TF_USE_CUDNN_BATCHNORM_SPATIAL_PERSISTENT'] = '0'
# Available since 1.12. issue#15874 # Available since 1.12. issue#15874
os.environ['TF_ENABLE_WHILE_V2'] = '1' # But they're sometimes buggy. We leave this decision to users.
os.environ['TF_ENABLE_COND_V2'] = '1' # os.environ['TF_ENABLE_WHILE_V2'] = '1'
# os.environ['TF_ENABLE_COND_V2'] = '1'
try: try:
import tensorflow as tf # noqa import tensorflow as tf # noqa
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment