作业系统:(ex:mac,win10,win8,win7,linux,etc...) Win10 问题类别:(ex：ML,SVM,RL,DL,RNN,CNN,NLP,BD,Vis,etc...) DL、CNN 使用工具:(ex:python,Java,C++,tensorflow,keras,MATLAB,etc...) tensorflow tensorflowsharp keras 问题内容: 各位版友大家好~ 目前使用一个叫tensorflowsharp(tf#)的工具，主要功能是可以直接在C#使用tensorflow 或keras训练出来的.pb格式模型 目前遇到的问题是，我先用keras训练完一个.h5的模型，再转成.pb格式， 并在C#程式中用tf#调用。 如果是用cpu运算基本上没问题，但如果改用GPU运算就会有下面的错误: Blas SGEMM launch failed:... https://imgur.com/a/lDKmTSJ 且这个BUG只发生在kernel size为1的卷积层 google後大家都说是因为显存被占用，但我有确认过我并没有开其他程式或 执行绪在使用显卡，且如果没有kernel size为1的卷积层就不会有这个bug 也就是说一样的模型，把卷积层的kernal size由1改为3就没问题。 请问有没有人解决过类似的问题呢? 本来怀疑是cudnn的bug，但一样的模型在python调用是没问题的， 只有在C#调用时才会有bug。 相关版本资讯: tensorflow:1.10.0 cuda=9.0.176 cudnn=7.3.1.20 tf#=1.7 以上，谢谢罗~ --

※ 发信站: 批踢踢实业坊(ptt.cc), 来自: 123.205.167.48 ※ 文章网址: https://webptt.com/cn.aspx?n=bbs/DataScience/M.1548942428.A.245.html