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

作者: jenny33377 (澄)   2017-10-11 22:30:21
※ 引述《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
作者: davis777 (大維斯)   2017-10-12 09:43:00
多卡可以用CUDA_VISIBLE_DEVICES="0"指定gpu
作者: hl4 (Zec)   2017-10-12 09:49:00
預設就是佔滿整個gpu 記憶體,後面的就會沒辦法用
作者: Kazimir (Kazimir)   2017-10-12 15:09:00
你看一下5600那篇 是不是你想要的
作者: Mchord (Mchord)   2017-10-12 21:49:00

Links booklink

Contact Us: admin [ a t ] ucptt.com