如何jupyter notebook上使用TensorFlow的系统报错问题

本篇文章给大家分享的是有关如何jupyter notebook上使用TensorFlow的系统报错问题，小编觉得挺实用的，因此分享给大家学习，希望大家阅读完这篇文章后可以有所收获，话不多说，跟着小编一起来看看吧。

CuDNNSLTM类型错误：找不到dnn实现。

解决方案：

import tensorflow as tf
from tensorflow import keras
config = tf.compat.v1.ConfigProto()
config.gpu_options.allow_growth = True
sess=tf.compat.v1.Session(config=config)
tf.compat.v1.keras.backend.set_session(sess)

未知错误：获取卷积算法失败。这可能是因为cuDNN未能初始化，所以请尝试查看上面是否打印了警告日志消息。

[[node functional_1/mobilenetv2_1.00_224/Conv1/Conv2D (defined at <ipython-input-44-7c882a77a6f7>:5) ]] [Op:__inference_train_function_15257] Function call stack: train_function

解决方案：

config = tf.compat.v1.ConfigProto(allow_soft_placement=True)
config.gpu_options.per_process_gpu_memory_fraction = 0.3
tf.compat.v1.keras.backend.set_session(tf.compat.v1.Session(config=config))

以上就是如何jupyter notebook上使用TensorFlow的系统报错问题，小编相信有部分知识点可能是我们日常工作会见到或用到的。希望你能通过这篇文章学到更多知识。更多详情敬请关注行业资讯频道。