问题处理failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED

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

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

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 虽然程序还是能正常运行,但是满满的红色对强迫症太不友好了
failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED
上网上找了很多解决方案,他们都说是什么动态显存没设置,我也去看了一下gpu的cuda占用,发现的确gpu的cuda占用拉满了。起初我以为是我电脑的显存太小了,运行不够,后来偶然间在解决另一个bug的时

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值