YOLO在本机运行时候报错 10C

问题描述:
InternalError (see above for traceback): Blas SGEMM launch failed : m=81920, n=32, k=64
[[node conv2d_3/convolution (defined at j:\ProgramData\Anaconda3\envs\tensorflow\lib\site-packages\keras\backend\tensorflow_backend.py:3650) = Conv2DT=DT_FLOAT, _class=["loc:@batch_normalization_3/cond/FusedBatchNorm/Switch"], data_format="NHWC", dilations=[1, 1, 1, 1], padding="SAME", strides=[1, 1, 1, 1], use_cudnn_on_gpu=true, _device="/job:localhost/replica:0/task:0/device:GPU:0"]]
[[{{node concat_11/_2897}} = _Recvclient_terminated=false, recv_device="/job:localhost/replica:0/task:0/device:CPU:0", send_device="/job:localhost/replica:0/task:0/device:GPU:0", send_device_incarnation=1, tensor_name="edge_3860_concat_11", tensor_type=DT_FLOAT, _device="/job:localhost/replica:0/task:0/device:CPU:0"]]

用了https://blog.csdn.net/gsww404/article/details/80507704上所说的方法还是无法解决该问题,请问还有可能是什么运行吗?

1个回答

用cpu试试看,怀疑是你gpu显存太小,爆显存了。

lvtao913
泠零〇 回复天空色的森林: 没解决,但是有个大概估计,因为你windows模式下是单显卡状态,要保证界面显示会把GPU内存占用,如果跑tensorflow可能会超负载,后来我再linux下没出现过这个问题。如果windows,我建议双显卡,一个做显示,一个跑tensorflow,应该就不会出问题(未经验证)
11 个月之前 回复
qq_37747186
天空色的森林 回复泠零〇: 我也出现了同样的问题,请问你解决了吗
11 个月之前 回复
lvtao913
泠零〇 回复caozhy: 我用session的config设置了最多只能用到4G的,用CPU跑tensorflow不是。。。,而且tesla P100的专业卡也就是16G吧,你说的48G都是多GPU模式吧
一年多之前 回复
caozhy
回答这么多问题就耍赖把我的积分一笔勾销了 回复泠零〇: 8GB未必够。一般专业卡都是24GB 48GB 甚至更多
一年多之前 回复
lvtao913
泠零〇 RTX2070,8G显存了,不小的
一年多之前 回复
Csdn user default icon
上传中...
上传图片
插入图片
抄袭、复制答案,以达到刷声望分或其他目的的行为,在CSDN问答是严格禁止的,一经发现立刻封号。是时候展现真正的技术了!
立即提问
相关内容推荐