Skip to content

Instantly share code, notes, and snippets.

@rhee-elten

rhee-elten/TF_FORCE_GPU_ALLOW_GROWTH_example.bashrc

Forked from rhee/TF_FORCE_GPU_ALLOW_GROWTH_example.bashrc
Last active May 16, 2022 03:12
Show Gist options
  • Save rhee-elten/22fed80fd95680ad68c000a7ea55a756 to your computer and use it in GitHub Desktop.
Save rhee-elten/22fed80fd95680ad68c000a7ea55a756 to your computer and use it in GitHub Desktop.
Download ZIP
TF_FORCE_GPU_ALLOW_GROWTH_example.bashrc
Raw
TF_FORCE_GPU_ALLOW_GROWTH_example.bashrc
:
export TF_FORCE_GPU_ALLOW_GROWTH=true
export CUDA_DEVICE_ORDER=PCI_BUS_ID
export CUDA_VISIBLE_DEVICES=0,1,2,3
export TF_CPP_MIN_LOG_LEVEL=3
# 참고: https://github.com/tensorflow/tensorflow/issues/22936#issuecomment-441790882
# Would you mind running your code with the following environment variables set:
# TF_ENABLE_WHILE_V2=1 TF_ENABLE_COND_V2=1 ??
export TF_ENABLE_WHILE_V2=1
export TF_ENABLE_COND_V2=1
export TF_GPU_HOST_MEM_LIMIT_IN_MB=65536 # (default, means 64GB)
export TF_DETERMINISTIC_OPS=1
TF_GPU_THREAD_MODE
TF_GPU_THREAD_COUNT
TF_USE_CUDNN_BATCHNORM_SPATIAL_PERSISTENT=1
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment