2021-04-06

tensorflow的一些bug系列

1.tensorflow.python.framework.errors_impl.UnknownError: 2 root error(s) found.

这个很容易让人觉得是cudnn版本问题，但是一会看cuda没毛病。实际上是gpu动态内存分配没写，代码用的是默认参数。下面的代码放在头部即可解决这个问题。

import os
os.environ["CUDA_VISIBLE_DEVICES"] = '0'
gpus = tf.config.experimental.list_physical_devices(device_type='GPU')
for gpu in gpus:
    tf.config.experimental.set_memory_growth(gpu, True) #gpu内存动态分配