作者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/cn.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