RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCre

最新推荐文章于 2024-08-01 21:22:04 发布
最新推荐文章于 2024-08-01 21:22:04 发布
阅读量8.1k 收藏 1
点赞数 3
分类专栏： debug python
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_41529093/article/details/118511415
版权
python 同时被 2 个专栏收录
453 篇文章 18 订阅
订阅专栏
debug
42 篇文章 0 订阅
订阅专栏

仅作为记录，大佬请跳过。

出现原因：

是模型的分类数不匹配

在这里插入图片描述
博主要进行四分类，但resnet18在这一行的代码（将最后一层全连接）设置成了三分类，因此error。

将3改为4，即可解决。

在这里插入图片描述

参考：传送门

注： 这个模型用的是迁移学习传送门

展示：

在这里插入图片描述

captain飞虎大队
关注
专栏目录
Python RuntimeError: thread.__init__() not called解决方法
09-22
然而，在使用`threading.Thread`进行多线程编程时，可能会遇到`RuntimeError: thread.__init__() not called`这样的错误。这个错误通常意味着你没有正确地初始化`threading.Thread`的实例。 在标题和描述中提到的...
回复上一篇文章，RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling cublasCreate(handle)成功解决
LamourL的博客
03-11 8043
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling cublasCreate(handle)
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`问题解决
不撸先疯。
07-02 1万+
使用transformers包调用pytorch框架的Bert预训练模型时，使用正常的bert-base-cased在其他数据集上正常运行，但是运用Roberta的时候却一直报错：RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`忙活了好几天也没查出是什么错误，网上一直提醒batch_size是否太大导致，修改为16->8->4->2都没有用。通过与其他数据集的对比，发现我在toke
【Pytorch】成功解决RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)
最新发布
创作高质量博文，分享知识，共同进步！
08-01 1651
🚀 遇PyTorch CUDA内存分配失败？别怕！一文带你从根源解决`CUBLAS_STATUS_ALLOC_FAILED`错误。通过检查显存、优化批处理、修改模型输出层及释放显存，轻松应对GPU内存挑战。实操示例助你上手快，让深度学习不再受阻！#PyTorch #CUDA错误 #内存优化 #深度学习
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`解决方案
qq_22721735的博客
08-16 4355
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`解决方案
解决错误：RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
qq_46080349的博客
10-23 3715
如标签的数量不等于网络输出通道的数量，即预测的输出类的数量。调整输出以匹配，它应该可以解决问题。1.减小batch_size。检查GPU内存，这个错误很可能是由于GPU的运行内存不够引起的。此外，可以使用如下命令格式来输出它的底层问题。5.上述方案都不能解决时候（2.更新GPU驱动程序。
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
qq_20373723的博客
03-20 2017
这个问题存在的其中一种可能就是模型的最后需要加sigmoid File ".\train.py", line 192, in train train_loss = solver.optimize() File "D:\wcs\seg_v2\framework.py", line 89, in optimize loss.backward() File "C:\Users\Administrator\Anaconda3\lib\site-packages\torch\tensor.p
Could not load dynamic library 'cublas64_11.dll';此类报错的dll资源
02-26
cudart64_110 cufft64_10 cublas64_11 cublasLt64_11 cusolver64_11 curand64_10 cusparse64_11 cudnn64_8等
解决docker报错Error response from daemon oci runtime error_centos安装docker
01-07
解决docker报错Error response from daemon oci runtime error
如何解决 RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
yaohaishen的专栏
01-13 1万+
本文方法限以下条件使用 运行环境没有改变 之前代码可以正常运行 代码修改后再同样的环境下出现了这个问题 如果满足以上三个条件，则一般不是CUDA出现了问题，则可以尝试用以下方法： 将代码在cpu上运行，不用GPU，这时候再报错就回给出具体出现问题的代码位置，有针对性的进行修改就好了 ...
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling ‘cublasCreate(handle)
qq_40390668的博客
12-19 3750
大多数问题是说显存爆了，如果不是显存爆了。我在做图像分类的时候，代码配置里需要手动配置图像的类别，如果类别数和你的数据集的类别数不一致，我就报了这个错。尝试检查类别数目是否设置正确。
python运行报错RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
热门推荐
m0_52347246的博客
09-08 3万+
python运行报错*** RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling cublasCreate(handle) ***解决办法： 将所设的batch_size改小一点。因为是从github上copy的代码，所以跑不通其实挺正常，毕竟电脑配置啥的，都与作者的不一样。因此我们改成和自己电脑相匹配的就行。 将batch_size改成32，就成功运行。 大家遇到这种情况根据自身情况，改小一点即可。 本文借鉴于https://
成功解决RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
QAQ6QAQ的博客
10-01 1056
这个错误实际上可能是batch_size过大而导致的，因为有些人可能是用的A100-80G显存，而有些人可能只有24G-4090或者3090。
pytorch模型训练遇到RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)
qq_20373723的博客
12-29 6016
具体错误如下： RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)` Exception raised from createCublasHandle at ..\aten\src\ATen\cuda\CublasHandlePool.cpp:8 (most recent call first): 00007FF937A175A200007FF937A17540 c10.dll!c10
pytorch报错：RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
科研学习笔记！
07-13 6156
多GPU分布式运行pytorch程序报错：RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling ‘cublasCreate(handle)`
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相关函数时出现了内存分配失败的错误。这可能是由于所用的显卡内存不足或者显存分配错误导致的。 为了解决这个问题，可以尝试以下方法： 1. 将所设的batch_size改小一点。减小batch_size可以减少显存的消耗，从而在显存有限的情况下减少内存分配失败的可能性。 2. 检查显卡的可用内存。确保显卡的内存足够大以容纳模型和数据的加载。 3. 确保使用的CUDA版本与所使用的深度学习框架和包兼容。某些版本的CUDA与特定版本的深度学习框架和包不兼容，可能导致错误的出现。 4. 如果问题仍然存在，可以尝试使用更高配置的显卡或者使用云端GPU来运行代码。
评论 2
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值