CUBLAS_STATUS_ALLOC_FAILED

最新推荐文章于 2023-03-04 15:32:30 发布

AI算法网奇

最新推荐文章于 2023-03-04 15:32:30 发布

阅读量2.3k

点赞数

CC 4.0 BY-SA版权

分类专栏： pytorch知识宝典

本文链接：https://blog.csdn.net/jacke121/article/details/114944568

pytorch知识宝典专栏收录该内容

512 篇文章 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

本文介绍了在使用CUDA时遇到的CUBLAS_STATUS_ALLOC_FAILED问题及其解决方案，主要内容包括如何通过检查和调整CUDA内存绑定来解决在模型预测时出现的错误。

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

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 should have the same spatial size as the output
>>> # of the convolution. In this case, for an input of 10, stride of 1
>>> # and kernel size of 3, without padding, the output size is 8
>>> offset = torch.rand(5, 2 * kh * kw, 8, 8)
>>> out = deform_conv2d(input, offset, weight)