运行莫烦TensorFlow教程报错,例子3
https://morvanzhou.github.io/tutorials/machine-learning/tensorflow/3-2-create-NN/
错误:
InternalError (see above for traceback): Blas GEMM launch failed : a.shape=(300, 1), b.shape=(1, 10), m=300, n=10, k=1
[[Node: MatMul = MatMul[T=DT_FLOAT, transpose_a=false, transpose_b=false, _device="/job:localhost/replica:0/task:0/gpu:0"](_arg_Placeholder_0_0/_7, Variable/read)]]
日志报错:
failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED
后查询资料
https://segmentfault.com/a/1190000009954640?utm_source=itdadao&utm_medium=referral
解决方案:
如果你是使用 GPU 版 TensorFlow 的话,并且你想在显卡高占用率的情况下(比如玩游戏)训练模型,那你要注意在初始化 Session 的时候为其分配固定数量的显存,否则可能会在开始训练的时候直接报错退出:
这时你需要用下面的方法创建 Session:
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.333)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
分享到:
相关推荐
注意:本资源是文章“tensorflow训练验证码识别模型”(https://blog.csdn.net/dcrmg/article/details/79005532#comments)的示例模型,分享的目的仅仅是为了让整个训练测试流程可以跑通,不保证准确率!不保证准确...
如何用Tensorflow训练模型成pb文件和和如何加载已经训练好的模型文件1
深度学习框架(TensorFlow)基础教程(全套PPT,代码以及素材)
今天小编就为大家分享一篇解决TensorFlow模型恢复报错的问题,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
深度学习框架(TensorFlow)基础教程(全套PPT,代码以及素材)
深度学习框架(TensorFlow)基础教程(全套PPT,代码以及素材)
Tensorflow训练好的模型,包含了各种调用方法讲解。
利用别人训练好的模型进行图像识别,实现Tensorflow快速入门!
深度学习框架(TensorFlow)基础教程(全套PPT,代码以及素材)
第二个是自己训练Tensorflow模型,训练几千次之后,将训练好的模型(文档中给出了详细的模型训练,具体训练的图片安装说明再准备提供即可使用),编译成APK,安装运行到Android客户端。适合初学者,能很快上手。具体...
深度学习框架(TensorFlow)基础教程(全套PPT,代码以及素材)
深度学习框架(TensorFlow)基础教程(全套PPT,代码以及素材)
深度学习框架(TensorFlow)基础教程(全套PPT,代码以及素材)
深度学习框架(TensorFlow)基础教程(全套PPT,代码以及素材)
深度学习框架(TensorFlow)基础教程(全套PPT,代码以及素材)
深度学习框架(TensorFlow)基础教程(全套PPT,代码以及素材)
深度学习框架(TensorFlow)基础教程(全套PPT,代码以及素材)
今天小编就为大家分享一篇Tensorflow加载预训练模型和保存模型的实例,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
在java中调用训练好的TensorFlow模型1
3、Tensorflow快速入门二 ——实战演练和模型训练 4、Tensorflow快速入门三 ——技巧总结 5、深度学习数学知识一览表 6、作业:实例:用自己的数据训练一个二分类模型 7、作业讲解:如何制作自己的数据集 第三...