chatbot
chatbot copied to clipboard
tensorflow.python.framework.errors_impl.UnknownError: Fail to find the dnn implementation.
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)