解决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快太多了。

你可能感兴趣的:(解决Nvidia CUDA安装后运行tensorflow报错InternalError (see above for traceback): Blas GEMM launch failed 20...)