问题
在训练到一定迭代次数之后报错:
RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)

可能的原因

  • shape维度不匹配
  • 变量不在同一个device上
  • pytorch和cuda版本不匹配

解决方案

在train.py文件的开头加上 os.environ['CUDA_VISIBLE_DEVICES'] = '0' ,并且设置device='cuda'
但是有 一个很奇怪的现象:如果不设置可见gpu,而是指定device='cuda:0'的话,也会报错。

【bug记录】RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm相关推荐

  1. RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm 解决方案

    目录 问题描述 解决方案 后记 问题描述 跑pytorch相关的代码时,在根据requirements.txt进行环境配置后,运行模型测试代码,出现了以下错误 File "/opt/cond ...

  2. RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm( handle, opa, opb

    这个问题是在测试数据时候遇到的,测试的数据是变长的,没有处理直接送进模型, 有看到说是一次性加载的数据太多(电脑不行),不过也确实一次性加载的数组是(1w+,1,512) 然后修改了一下dataloa ...

  3. 报错RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasGemmEx( handle, opa,

    RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling cublasGemmEx( handle, opa, opb ...

  4. CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm

    CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm 在模型测试的时候出现了这个问题, CUDA error: CU ...

  5. 【PyTorch】RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling cublasSgemm()

    完整报错信息. RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling `cublasSgemm( handle, opa ...

  6. RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling `cublasSgemm( handle, opa, opb, m

    RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling cublasSgemm( handle, opa, opb, m, ...

  7. RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`

    RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling cublasCreate(handle) 在写pytorch情感分类 ...

  8. RuntimeError: CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling `cublas‘

    调用nn.linear时出现RuntimeError: CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling `cublas'错误,搜索网上资料 ...

  9. CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling `cublasSgemm( handle, opa, opb, m, n, k, alpha

    https://blog.csdn.net/u014176855/article/details/80994991 他的方法对我这个不奏效.检查发现是分类网络的全连接layer第一层神经元的数量设置错 ...

最新文章

  1. 性能优化指南:性能优化的一般性原则与方法
  2. FPGA从Xilinx 的7系列学起(5)
  3. 阻塞队列的应用 || 多线程的条件判断 一定要用while,而不要用 if
  4. 中单引号和双引号的区别与联系_VB中Sub与function的联系与区别
  5. Python将秒转换为 时分秒 的格式
  6. 1_2 AbstractFactoryMode 抽象工厂模式
  7. 浅析Entity Framework Core中的并发处理
  8. SharePoint中的权限体系
  9. 解决输入框自动填充账号密码的问题
  10. android多文件上传错误,在Android中上传大文件而不会出现内存错误
  11. 前端实现图片或视频预览的3种方法
  12. 无法自动修复此计算机鼠标按不了机,win10系统罗技M558鼠标无法自动连接的修复方法...
  13. 如何在 Spring Boot 优雅关闭加入一些自定义机制
  14. 又整理了一些面试相关的:视频教程,面试经验,简历模板,写简历的技巧等
  15. Python免安装环境(Windows)
  16. 使用Python调用opencv学习(-)打开图片,显示图片
  17. Windows系统的四个常见后门
  18. 搜索引擎排名都选乐云seo_百度官网排名-搜索引擎算法优化
  19. matlab进行动力吸振器设计,干摩擦动力吸振器的MATLAB仿真及其基于频谱分析的优化设计...
  20. 设计模式---状态模式(State Pattern)

热门文章

  1. linux安装tune2fs工具,linux tune2fs命令详解
  2. Unity中的Shuriken粒子系统(5)
  3. GitHub代码阅读神器,你值有拥有!
  4. 初探MITM-中间人攻击
  5. Red hat Linux 重启、关机、退出X、启动X 命令
  6. 高通dsps部分的打印
  7. 项目8 利用DHCP自动分配IP地址
  8. 又一行业盛会落地深圳!世链“FAF世界区块链3000人大会”全网热度领先
  9. 摄影师选图脚本(根据快照文件的存在与否整理原文件)
  10. job vacancy