[問題]TensorFlowSharp 用gpu運算時的BUG

[chchan1111 (123)]

作業系統:(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://www.ptt.cc/bbs/DataScience/M.1548942428.A.245.html
→ benson415: And your data types are? Float32 should be fine.
推 maxsho: 也有可能是c#套件本身的bug，先goo看看是否有類似情況