chatbot icon indicating copy to clipboard operation
chatbot copied to clipboard

tensorflow.python.framework.errors_impl.UnknownError: Fail to find the dnn implementation.

Open terio2020 opened this issue 3 years ago • 0 comments

cudnn分配不出更多的运算资源了

添加

import tensorflow as tf gpus = tf.config.experimental.list_physical_devices('GPU') for gpu in gpus: tf.config.experimental.set_memory_growth(gpu, True)

terio2020 avatar Jun 04 '21 14:06 terio2020