failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED 报警告问题的解决
MY question:总是抱这个错,但是我的gpu并没有被占用,找了一些方法,网上最多的那种方法总是提示ConfigProto有问题
然后找到了下面的解决方法,虽然也会跳红色的

但是最后不会提示显存不足,会正常训练!如果有谁知道这个红色的部分怎么解决,望告知,谢谢!
转自:https://www.tqwba.com/x_d/jishu/202284.html
failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED 报警告问题的解决
最近在跑keras的时候,发现配置好的cuda和cudnn,在tensorflow-gpu 2.3.0的情况下,老是会报出一个警告failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED 虽然程序还是能正常运行,但是满满的红色对强迫症太不友好了
上网上找了很多解决方案,他们都说是什么动态显存没设置,我也去看了一下gpu的cuda占用,发现的确gpu的cuda占用拉满了。起初我以为是我电脑的显存太小了,运行不够,后来偶然间在解决另一个bug的时

当运行TensorFlow-GPU时遇到'failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED'警告,即使GPU未被占用,本文提供了解决方案。通过在程序开头添加特定代码,可以消除警告并优化显存使用。

5943

被折叠的 条评论
为什么被折叠?



