解决Nvidia CUDA安装后运行tensorflow报错InternalError (see above for traceback): Blas GEMM launch failed 20...

代码是之前在cpu版本的tensorflow上正常运行的,昨天装上CUDA和GPU版本的tf,马上试着跑一下,报错。内容如下:

InternalError (see above for traceback): Blas GEMM launch failed

有人说显卡被其他应用占用造成内存不够,靠配置重新制定CUDA显卡内存分配,我觉得这事应该没这么复杂,第一我用Linux,装CUDA的时候设置--no-opengl-files,第二,本来也没什么吃显卡的应用。
相信大厂,相信不会这么给用户添麻烦。打开的tab里发现测试的时候有个tf的session开了没关,应该就是他了,shutdown 所有在运行的python kernel,clear and run all, 一切正常。

GeForce GTX 760M 的GPU跑一个3层每层2000个神经元的简单网络,比我15年的i5CPU的MBP快6倍,6倍,6倍,有条件还是要上GPU,而且不用太好的显卡,我这个笔记本上的7系列老显卡官方给出Compute Capability是3.0,9系列是5,1000系列大概是6,可比cpu快太多了。

最后编辑于 :
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容