Re: [問題] 用keras模組時,要怎麼用cuda和cudnn

看板Python作者 (澄)時間6年前 (2017/10/11 22:30), 6年前編輯推噓2(202)
留言4則, 4人參與, 6年前最新討論串2/2 (看更多)
※ 引述《ssd860505da (JAGER)》之銘言: : 大家好,最近在學deep learning : 我用的環境是windows10 : 剛把keras弄好,backend:tensorflow : 我的CUDA和cudnn都載好了,環境變數裡也加上了 : 我CUDA的路徑是:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0 : cudnn的三個file已分別放入bin/include/lib裡 : 是不是還要再加個path, tensorflow才會知道要用顯卡來運算? : 我看CPU使用率100%,心都在淌血Q^Q : 我發現載了tensorflow-gpu的模組就會有bug : 我用的版本是Python 3.5.3 : 最後再問個, 用linux環境算這個會不會比較好 各位好,我今年才開始接觸Python 有幸看到這篇文章說明tensorflow gpu安裝方法 才順利成功利用gpu進行運算 接著 由於需要跑多個不同數據case情況,發現跑第一個case是可以執行gpu的 但果要跑第二個以後就不能跑了T_T 想詢問Keras是不能同時運作多個.py檔案,進行GPU運算嗎? 因為我的case有點多,如果用cpu是可以同時跑11個case, 只是cpu也100%滿滿der -- ※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 114.46.128.182 ※ 文章網址: https://www.ptt.cc/bbs/Python/M.1507732223.A.C7D.html

10/12 09:43, 6年前 , 1F
多卡可以用CUDA_VISIBLE_DEVICES="0"指定gpu
10/12 09:43, 1F

10/12 09:49, 6年前 , 2F
預設就是佔滿整個gpu 記憶體,後面的就會沒辦法用
10/12 09:49, 2F

10/12 15:09, 6年前 , 3F
你看一下5600那篇 是不是你想要的
10/12 15:09, 3F
我看了一下, 只是說我程式是用Keras寫的,只是backend是使用tensorflow Keras和tenserflow可以互相混合寫嗎?? ※ 編輯: jenny33377 (114.46.128.182), 10/12/2017 21:00:24 ※ 編輯: jenny33377 (114.46.128.182), 10/12/2017 21:00:59

10/12 21:49, 6年前 , 4F
10/12 21:49, 4F
感謝M大的方法,已經可以跑了 也謝謝各位前輩的幫助 ※ 編輯: jenny33377 (114.46.128.182), 10/13/2017 08:32:21
文章代碼(AID): #1PtYh_nz (Python)
文章代碼(AID): #1PtYh_nz (Python)