作者ssd860505da (JAGER)
看板Python
標題[問題] 用keras模組時,要怎麼用cuda和cudnn
時間Wed Aug 23 17:41:11 2017
大家好,最近在學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環境算這個會不會比較好
--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 114.44.247.31
※ 文章網址: https://webptt.com/m.aspx?n=bbs/Python/M.1503481275.A.41A.html
1F:推 leoloveivy: cudnn先覆蓋進去 然後裝tensorflow gpu 你backend是gp08/23 19:16
2F:→ leoloveivy: u version keras不用特別調08/23 19:16
3F:→ ssd860505da: 什麼意思,我cudnn的檔案已經放進去了 ,可我裝了tens08/23 20:02
4F:→ ssd860505da: orflow-gpu之後執行就會有bug08/23 20:02
5F:推 Kazimir: 我只有用過tensorflow 不過我的經驗是你只要把cudnn擺對08/23 20:43
6F:→ Kazimir: 位置就OK 你說的有bug是什麼意思? import 會成功嗎?08/23 20:44
7F:推 Kazimir: tf.test.gpu_device_name() 有東西嗎? 08/23 20:48
這是裝了tensorflow-gpu之後才有的error
https://pastebin.com/raw/3pCqMva0
8F:→ f496328mm: 我是用 ubuntu 環境 都很難裝就是了XD08/23 22:24
9F:→ f496328mm: 如果你不熟悉 linux 會更難 有一堆雷08/23 22:25
10F:推 Kazimir: 你可能需要確認一下tensorflow cuda cudnn彼此的支援版本08/24 20:48
11F:推 Kazimir: 你可以試試建一個env 灌tf看看會不會動 我剛剛灌這台08/24 22:02
12F:→ Kazimir: 有一些和你類似的問題 重灌可能會有效08/24 22:02
13F:推 Kazimir: 還要確定有灌visual c++ 201508/24 23:43
windows10有灌,cuda8.0, cudnn5.1,位置應該有放對,環境變數path有加cuda
※ 編輯: ssd860505da (180.217.150.255), 08/25/2017 17:38:24
14F:推 EGsux: 你GPU是哪一個 08/26 03:01
15F:→ ssd860505da: NVIDIA Coporation GK208M [GeForce 920M] 08/26 03:14
16F:推 Kazimir: 你的tensorflow是幾版的? 1.3是用cudnn 6 喔 08/27 06:44
我改成cudnn 6之後可以執行成功,CPU的使用率降到40%左右,可是會出現這段訊息
https://pastebin.com/raw/Gj7724Wv
這樣算成功嗎?
※ 編輯: ssd860505da (180.217.143.218), 08/27/2017 12:50:43
17F:推 Kazimir: 恩 這樣OK了 就像他說的你的顯卡不完全適合幹這個 08/27 12:53
18F:→ Kazimir: 不過2GB的卡本來就只能玩玩而已XD 08/27 12:53
19F:推 Kazimir: 啊 抱歉 我在胡說八道 tensorflow可以使用avx指令集 08/27 13:03
20F:→ Kazimir: 但是pip來的版本不能 所以假設使用cpu訓練的話 改善這個 08/27 13:05
21F:→ Kazimir: 問題可以加快學習時間 不過用GPU的話就沒差了 08/27 13:06
23F:→ z115988: 路過問一下,請問關於機器學習,PTT有相關版嗎?=3= 08/27 23:33
24F:→ ssd860505da: 等你開阿=3= 08/28 00:41
25F:→ ssd860505da: 就是找不到這種版所以只好來這問 08/28 00:42
26F:推 f496328mm: ML基本上就 R or Python 來這兩個版問就好了 08/29 01:16
27F:→ f496328mm: 厲害一點的 Java or C++ 也是有人在做 08/29 01:16
28F:推 p00o99o: 他上面哪有寫卡不適合... 09/10 23:22
29F:推 leoloveivy: 運算<3.0不行 所以有一些人也卡在這 10/13 15:15