【改bug】——RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`

最新推荐文章于 2024-08-01 21:22:04 发布
最新推荐文章于 2024-08-01 21:22:04 发布
阅读量5.8k 收藏 4
点赞数 3
分类专栏： bug记录 文章标签： 神经网络 深度学习 pytorch
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_41488595/article/details/121547300
版权
在运行VGG模型时遇到bug，发现是由于CUDA版本11.0与安装的PyTorch版本11.1不兼容。通过查找解决方案并回退到旧版CUDA，最终成功解决了问题。博客作者分享了在Windows环境下直接配置环境的经验，并以此记录解决问题的过程。
摘要由CSDN通过智能技术生成

今天在跑vgg的时候碰到了这个bug，因为之前的模型都没有碰到这个bug，也一时无了头绪，就去网上找解决办法
这个老哥碰到的是分类数目与模型最后输出的数目不同
在这里插入图片描述
这个问题不是我的问题，下一个
这个老哥碰到的是有一个地方没有添加to(device)
在这里插入图片描述
这个问题不是我的问题，下一个
然后的解决办法就大多一样了，一直没有解决

求助万能的群友去了
在这里插入图片描述
在这里插入图片描述

去看了看，发现自己cuda是11.0版本，安装的torch是11.1cuda版本
在这里插入图片描述
在这里插入图片描述
之后去找了找老版本torch下载地址
在这里插入图片描述
安装，搞定（因为我这个电脑显卡比较垃圾，没有装ubuntu，还是windows，平时学习用这个电脑，也就没有装conda，直接在裸机上配环境）

仅以这个文章纪念这个bug

每个人都是孙笑川
关注
专栏目录
cuda_setup.add_log_entry(“WARNING: Compute capability ＜ 7.5 detected解决方案
weixin_43178406的博客
04-13 8万+
本文主要介绍了cuda_setup.add_log_entry(“WARNING: Compute capability ＜ 7.5 detected解决方案，希望能对使用GPU的同学有所帮助。 文章目录 1. 问题描述 2. 解决方案
pytorch报错：RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
科研学习笔记！
07-13 6158
多GPU分布式运行pytorch程序报错：RuntimeError: 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.
CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
Saulooo的博客
06-11 4365
CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
【Pytorch】成功解决RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)
最新发布
创作高质量博文，分享知识，共同进步！
08-01 1671
🚀 遇PyTorch CUDA内存分配失败？别怕！一文带你从根源解决`CUBLAS_STATUS_ALLOC_FAILED`错误。通过检查显存、优化批处理、修改模型输出层及释放显存，轻松应对GPU内存挑战。实操示例助你上手快，让深度学习不再受阻！#PyTorch #CUDA错误 #内存优化 #深度学习
（报错解决）RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasGemmEx( handle, o
Mmaa100的博客
04-01 765
当时我还在冥思苦想为什么明明运行了cuda安装对应指令，但在查看cuda是否可用时，仍输出为False，后来才知道是系统默认安装了cpu版本的库。该报错我搜集了很多资料，给出的解答很多，其实实际上就是cuda版本问题，对于pytorch1.8.1而言，cuda至少要11.x版本以上才行，当然其他pytorch版本遇到了这个报错，也建议将cuda安装到11.x版本以上。另外，在pytorch官网匹配版本时，当我使用conda安装11.x版本以上的cuda时，即。再运行代码，报错解决。
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
weixin_44357285的博客
03-20 2214
CUBLAS 是 CUDA 专门用来解决线性代数运算的库。而上述报错则是说明读取的数据类别数与网络模型中设置的类别数不一致造成。比如，你的数据是5各类别，而你的网络中的分类数（num_classes)设置成了6。首先，你要检查读取数据的路径看是否是正确读取想要的数据；对比是否labels的最大值与读取数据的类别数一致。然后，打印网络中的标签（即labels)；
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling ‘cublasCreate(handle)
qq_40390668的博客
12-19 3750
大多数问题是说显存爆了，如果不是显存爆了。我在做图像分类的时候，代码配置里需要手动配置图像的类别，如果类别数和你的数据集的类别数不一致，我就报了这个错。尝试检查类别数目是否设置正确。
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.电脑性能不行 ...
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)【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跑就不行了
数据集错误踩坑：CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
weixin_44119512的博客
09-29 2611
CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling cublasCreate(handle) 如果你遇到这个问题，并且在数据集里面看完全没问题，为什么就是报这个错误呢？ 很可能是你在数据集里打开了jupyter notebook 简 单统计数据了，那原因就是！打开了这个之后会隐藏一个.ipynb_checkpoints 文件夹 是一个与文件版本控制保存机制有关的东西。 ['cane', 'cavallo', 'elefante', 'farfalla',
模型测试时，调用bert 报错 CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
qq_40130606的博客
05-30 464
模型测试时，调用bert 报错 CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
预测时遇到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或者输入图像的分辨率，以减少GPU内存的使用。 2. 禁用梯度计算，以减少GPU内存的使用。可以在代码中添加以下语句来禁用梯度计算： ```python with torch.no_grad(): # your prediction code here ``` 3. 如果你的GPU内存足够大，可以尝试使用`CUDA_LAUNCH_BLOCKING=1`命令来输出底层问题。具体来说，可以使用以下命令格式来运行你的代码： ```bash CUDA_LAUNCH_BLOCKING=1 python [你的命令参数] ``` 例如： ```bash CUDA_LAUNCH_BLOCKING=1 python predict.py --task davis ``` 这将会输出更详细的错误信息，帮助你更好地定位问题。
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值