python运行报错RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`

最新推荐文章于 2024-08-18 02:54:03 发布
最新推荐文章于 2024-08-18 02:54:03 发布
阅读量3.2w 收藏 14
点赞数 13
文章标签： python 深度学习 剪枝
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/m0_52347246/article/details/120177149
版权
这篇博客介绍了在运行Python代码时遇到CUDA错误CUBLAS_STATUS_ALLOC_FAILED的解决方法。主要建议是调整batch_size至适合个人电脑配置的大小，例如将batch_size改为32，同时提醒用户根据数据集类别数量设置正确的num_classes。通过这些调整，可以解决由于资源分配问题导致的运行失败。
摘要由CSDN通过智能技术生成

python运行报错***

RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling cublasCreate(handle)

***解决办法：
1、将所设的batch_size改小一点。因为是从github上copy的代码，所以跑不通其实挺正常，毕竟电脑配置啥的，都与作者的不一样。因此我们改成和自己电脑相匹配的就行。
在这里插入图片描述
将batch_size改成32，就成功运行。
大家遇到这种情况根据自身情况，改小一点即可。

2、根据自己的数据集改成相应的num_classes.
eg.你的数据集为CIFAR100，有100类，则num_classes=100。

柠檬树下你和我₰
关注
pytorch报错：RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
科研学习笔记！
07-13 6162
多GPU分布式运行pytorch程序报错：RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling ‘cublasCreate(handle)`
RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling `cublasGemmEx 解决方案
热门推荐
weixin_43178406的博客
06-12 8万+
本文主要介绍了RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling `cublasGemmEx 解决方案，希望能对学习pytorch的同学们有所帮助。 文章目录 1. 问题描述 2. 解决方案
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling ‘cublasCreate(handle)
qq_40390668的博客
12-19 3750
大多数问题是说显存爆了，如果不是显存爆了。我在做图像分类的时候，代码配置里需要手动配置图像的类别，如果类别数和你的数据集的类别数不一致，我就报了这个错。尝试检查类别数目是否设置正确。
RuntimeError: CUDA error: operation not supported when calling cusparseCreate(handle)
最新发布
m0_52261601的博客
08-18 686
复现lightGCN在配置环境的时候出现了很多问题，狠狠记录让我恼火的几个大坑，在未来配置环境出现问题时提供一些思路（可恶autodl还我10块钱啊啊啊啊）：问题1根据官方配置的情况下出现很多安装包不匹配问题，官方配置如下（https://github.com/gusye1234/LightGCN-PyTorch/blob/master/requirements.txt）问题2当问题1解决之后，cuda持续报错。
CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
Saulooo的博客
06-11 4366
CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
【新方案】RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
懒惰是科技进步的原始动力
06-23 6371
CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm` CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)` CUDA error: device-side assert triggered IndexError: Target -1 is out of bounds.
pytorch:RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
长枫sky的博客
10-27 2万+
pytorch出现报错： RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling cublasCreate(handle) 将batch_size改小一点即可 我之前的 batch_size = 32, 改为 batch_size = 8 程序跑成功了。 出现上述问题的原因： 1.网络过于复杂（深） 2.电脑性能不行 ...
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
weixin_44357285的博客
03-20 2218
CUBLAS 是 CUDA 专门用来解决线性代数运算的库。而上述报错则是说明读取的数据类别数与网络模型中设置的类别数不一致造成。比如，你的数据是5各类别，而你的网络中的分类数（num_classes)设置成了6。首先，你要检查读取数据的路径看是否是正确读取想要的数据；对比是否labels的最大值与读取数据的类别数一致。然后，打印网络中的标签（即labels)；
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)
10-04
引用和引用[2]中提到的错误信息 "RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`" 表明在调用CUDA相关函数时出现了内存分配失败的错误。这可能是由于所用的显卡内存...
预测时遇到RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
12-04
当预测时遇到`RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling 'cublasCreate(handle)'`错误时，这通常是由于GPU内存不足导致的。可以尝试以下几种方法来解决这个问题： 1. 减少batch size或者...
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)` NLLLoss
09-19
当你在GPU上运行深度学习训练模型的程序时，报错"RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`"通常表示CUDA内存不足。这个错误可能是由于你的数据集扩充导致的。 ...
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling ‘cublasCreate(handle)【pytorch err】
weixin_43537379的博客
05-20 2720
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling 'cublasCreate(handle) 问题描述 使用 pytorch 跑程序 环境没变，换了代码就报了这个错误 RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling 'cublasCreate(handle)，尝试使用CPU运行，发现是可以跑通的，没有提示什么问题，但是换成GPU跑就不行了
解决错误：RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
qq_46080349的博客
10-23 3723
如标签的数量不等于网络输出通道的数量，即预测的输出类的数量。调整输出以匹配，它应该可以解决问题。1.减小batch_size。检查GPU内存，这个错误很可能是由于GPU的运行内存不够引起的。此外，可以使用如下命令格式来输出它的底层问题。5.上述方案都不能解决时候（2.更新GPU驱动程序。
【改bug】——RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
qq_41488595的博客
11-25 5803
今天在跑vgg的时候碰到了这个bug，因为之前的模型都没有碰到这个bug，也一时无了头绪，就去网上找解决办法 这个老哥碰到的是分类数目与模型最后输出的数目不同 这个问题不是我的问题，下一个 这个老哥碰到的是有一个地方没有添加to(device) 这个问题不是我的问题，下一个 然后的解决办法就大多一样了，一直没有解决 求助万能的群友去了 去看了看，发现自己cuda是11.0版本，安装的torch是11.1cuda版本 之后去找了找老版本torch下载地址 安装，搞定（因为我这个电脑显卡比较垃圾，
评论 4
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值