今天中午出发去井冈山参加创客培训,看培训的内容有python人工智能、机器人编程、3d打印、vr等内容,正好最近研究这方面,所以去学习一下还是不错的。昨天晚上搞到好晚,因为新笔记本电脑在用cpu训练tensorflow的项目时竟然卡机,然后任务管理器也调用不出来,而同样的项目在苹果笔记本电脑win8系统i5cpu下训练速度非常的慢,但不至于卡死。这样我遇到了两难的选择,最后我决定尝试用gpu来训练,但好像直接用代码调用gpu没有任何效果。于是我安装了tensorflow-gpu版本,但运行时又出现找不到cunnd64-7.dll,然后我根据网上的资料,安装了cuda,但第一次安装了cuda9.2,还是报错,然后我又安装cuda8.0,依然错误!然后我又折腾设置环境变量,重启电脑等都不行!后来网上资料说tf和cuda的版本要匹配,否则也不行.然后我就发现错误提示安装cuda9.0,但反复安装了几次都不成功,用鲁大师卸载现有gtx1060驱动也不行,然后网上搜索cuda9.0安装失败,才找到正确的方法,原来是visual vis什么安装不了,要选择自定义安装,然后去掉这个visual什么的,最后顺利安装上!然后又提示cudnn错误,下载了6.0版本没有用,官网上要注册用户登陆才能下载,然后从网上搜索其它下载链接如百度网盘,下载9.0版本,还是不能用,原来要把里面的cuda拷贝到cuda9.0安装目录下,然后设置环境变量,最终可以正常运行tf的gpu版本!但新问题又来了,我发现gpu版本其实启动很慢,启动后训练速度是更快了,但这个启动慢的问题网上说本来就是这样的,但相比耗时训练来说还是可以忽略的,不耗时的就cpu快了!
终于装上了tenflow的gpu版本并且正常运行
最新推荐文章于 2023-11-01 11:30:37 发布