[问题] 用keras模组时,要怎么用cuda和cudnn

楼主: ssd860505da (JAGER)   2017-08-23 17:41:11
大家好,最近在学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环境算这个会不会比较好
作者: leoloveivy (cried)   2017-08-23 19:16:00
cudnn先覆蓋进去 然后装tensorflow gpu 你backend是gpu version keras不用特别调
楼主: ssd860505da (JAGER)   2017-08-23 20:02:00
什么意思,我cudnn的档案已经放进去了 ,可我装了tensorflow-gpu之后执行就会有bug
作者: Kazimir (Kazimir)   2017-08-23 20:43:00
我只有用过tensorflow 不过我的经验是你只要把cudnn摆对位置就OK 你说的有bug是什么意思? import 会成功吗?tf.test.gpu_device_name() 有东西吗?
作者: f496328mm (为什么会流泪)   2017-08-23 22:24:00
我是用 ubuntu 环境 都很难装就是了XD如果你不熟悉 linux 会更难 有一堆雷
作者: Kazimir (Kazimir)   2017-08-24 20:48:00
你可能需要确认一下tensorflow cuda cudnn彼此的支援版本你可以试试建一个env 灌tf看看会不会动 我刚刚灌这台有一些和你类似的问题 重灌可能会有效还要确定有灌visual c++ 2015
作者: EGsux (天龙人)   2017-08-26 03:01:00
你GPU是哪一个
楼主: ssd860505da (JAGER)   2017-08-26 03:14:00
NVIDIA Coporation GK208M [GeForce 920M]
作者: Kazimir (Kazimir)   2017-08-27 06:44:00
你的tensorflow是几版的? 1.3是用cudnn 6 喔恩 这样OK了 就像他说的你的显卡不完全适合干这个不过2GB的卡本来就只能玩玩而已XD啊 抱歉 我在胡说八道 tensorflow可以使用avx指令集但是pip来的版本不能 所以假设使用cpu训练的话 改善这个问题可以加快学习时间 不过用GPU的话就没差了SO解释:https://goo.gl/XLtpxR
作者: z115988 (Lee)   2017-08-27 23:33:00
路过问一下,请问关于机器学习,PTT有相关版吗?=3=
楼主: ssd860505da (JAGER)   2017-08-28 00:41:00
等你开阿=3=就是找不到这种版所以只好来这问
作者: f496328mm (为什么会流泪)   2017-08-29 01:16:00
ML基本上就 R or Python 来这两个版问就好了厉害一点的 Java or C++ 也是有人在做

Links booklink

Contact Us: admin [ a t ] ucptt.com