CUBLAS_STATUS_ALLOC_FAILED 解决办法

最新推荐文章于 2024-07-11 14:35:56 发布
最新推荐文章于 2024-07-11 14:35:56 发布
阅读量1.7w 收藏 9
点赞数 5
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/zhangyumengs/article/details/80551633
版权

训练facenet中的train_softmax.py时，出现了以下错误

failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED

这个是由于GPU的配置不对造成的。

解决办法：

config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config, ...)


zhangyumengs
关注
  • 5
    点赞
  • 踩
  • 9
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
CUBLAS_STATUS_ALLOC_FAILED
jacke121的专栏
03-17 2259
CUBLAS_STATUS_ALLOC_FAILED 解决方法： 提取绑定cuda torch.rand(1, 3, 10, 10).cuda(7) 在训练时ok，只预测时结果不对。 >>> input = torch.rand(1, 3, 10, 10) >>> kh, kw = 3, 3 >>> weight = torch.rand(5, 3, kh, kw) >>> # offset shou...
pytorch报错：RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
科研学习笔记！
07-13 6083
多GPU分布式运行pytorch程序报错：RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling ‘cublasCreate(handle)`
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
最新发布
qq-1343330453
07-11 265
bert模型报错
CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
qq_38178543的博客
01-08 1670
我出现过两次这个问题，原因分别来自于： ** embedding维度问题 词表问题 ** 我怎么找出来的原因？ 我把它放在cpu上运行，就会报错误的具体位置了，而不是报如题所示的这种很迷惑的错误。 举例1 torch.embedding import torch import torch.nn as nn position_embeddings = nn.Embedding(10, 128)#10在这里是边界，不是维度 position_ids = torch.arange(1, 11, dtype
failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED
xiongzai2016的博客
11-26 2430
failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED 错误解决方法
failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED
WEXIA666的博客
09-02 2139
failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED问题的解决 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
smallwhite620451的博客
03-11 887
failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED 报警告问题的解决 MY question：总是抱这个错，但是我的gpu并没有被占用，找了一些方法，网上最多的那种方法总是提示ConfigProto有问题 然后找到了下面的解决方法，虽然也会跳红色的 但是最后不会提示显存不足，会正常训练！如果有谁知道这个红色的部分怎么解决，望告知，谢谢！ 转自：https://www.tqwba.com/x_d/jishu/202284.html
TensorFlow—GPU版本运行报错：failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED
锅盖小郭哥
02-23 2988
报错： 2021-02-23 00:00:47.646531: E tensorflow/stream_executor/cuda/cuda_blas.cc:238] failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED 原因： 此时说明你电脑显存占用过高，已经不足！! 如果继续 全负载跑比较复杂的TensorFlow网络模型则会导致电脑其他软件运行失败，起到保护电脑的作用。 解决： 需要在初始化 Session 的时候为其分配固定数量
【keras】CUDNN_STATUS_ALLOC_FAILED 和 failed to run cuBLAS routine cublas ，keras和tensflow-gpu版本的问题
qq_37791134的博客
03-01 2165
【报错】首先报错： could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED尝试解决： https://blog.csdn.net/weixin_44754046/article/details/97663626 在最开头添加： import tensorflow as tf config = tf.ConfigProto(gpu_o...
解决：failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED
Amigo_1997的博客
03-26 8497
今天测试tensorflow的GPU版本，但是发现一个问题同一个程序（卷积神经网络）为什么在anaconda里的jupyter里就可以飞速跑完，在使用anaconda环境的pycharm里面就会说显存不足呢？ 报错是：failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED（这就是显存不足的意思） 两个解决方案： （1） ...
failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED 报错 （解决方法）
Welcome
11-16 1969
1.报错 问题：（来自GCN_Keras-master代码） 在Pycharm运行GCN_Keras代码的时候，报错： ①Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA 您的CPU支持该TensorFlow二进制文件未编译为使用的指令：AVX2 FMA 原因：成功解决Your CPU supports instructions that this TensorF.
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
讲解CUBLAS_STATUS_ALLOC_FAILED CUDA_ERROR_OUT_OF_MEMORY
牛肉胡辣汤
01-03 2945
和是在使用CUDA和CUBLAS库时经常遇到的错误。它们都与内存分配有关，前者表示CUBLAS库的内存分配失败，后者表示CUDA运行时库的设备内存不足。解决这些错误的方法包括确保系统中有足够的可用内存、减小内存分配请求的大小、优化内存使用以及使用更高容量或更先进的GPU设备等。在开发使用CUDA和CUBLAS库的程序时，了解这些错误的原因和解决方法将有助于更好地处理内存分配和管理，以提高计算性能和可靠性。希望本文对您理解和解决CUBLAS和CUDA的内存分配问题有所帮助。
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCre
weixin_41529093的博客
07-06 8108
仅作为记录，大佬请跳过。 出现原因： 是模型的分类数不匹配 博主要进行四分类，但resnet18在这一行的代码（将最后一层全连接）设置成了三分类，因此error。 将3改为4，即可解决。
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling ‘cublasCreate(handle)
qq_40390668的博客
12-19 3746
大多数问题是说显存爆了，如果不是显存爆了。我在做图像分类的时候，代码配置里需要手动配置图像的类别，如果类别数和你的数据集的类别数不一致，我就报了这个错。尝试检查类别数目是否设置正确。
评论 2
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值