首页 > 技术文章 > Blas GEMV launch failed: m=3, n=10000

yinyoupoet 2019-12-23 22:40 原文

在使用tensorflow处理图片的时候,出现了如题所示错误。本次运行使用了独显进行计算,而错误原因是因为同时运行了多个占用显卡运行资源的程序!
比如同时运行了好几个jupyter的kernel,或者在pycharm中选择了run with python console且同时开启了多个console窗口,或者在运行jupyter的时候又运行了pycharm,就会造成这种问题。
因此解决办法就是:只保留一个占用显卡的进程!

至于原因,是因为tensorflow在运行时默认会占用所有剩余的显存,这样做是为了防止显存的存储空间碎片化,所以默认情况下每次只允许存在一个tensorflow进程,具体原因和解决方案可以看这篇文章:keras or tensorflow 显存占用问题

推荐阅读