fix for Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

gistfile1.txt

#
# Look at https://www.tensorflow.org/guide/gpu for an explaination
# Here are the two options they presented
#

# Option 1
# set_memory_growth before creatingt the CONV network

import tensorflow as tf
gpus = tf.config.experimental.list_physical_devices('GPU')
for gpu in gpus:
    tf.config.experimental.set_memory_growth(gpu, True)
    
# Option 2
# set env variable in shell or .bashrc

export TF_FORCE_GPU_ALLOW_GROWTH="true"
export TF_CPP_MIN_LOG_LEVEL=2