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

看板DataScience作者chchan1111 (123)時間5年前 (2019/01/31 21:47)推噓1(1推 0噓 1→)

留言2則, 2人參與討論串1/1

作業系統:(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