本文介绍: RuntimeError: cublas runtime error : the GPU program failed to execute at /tmp/pip–req-build-jh50bw28/aten/src/THC/THCBlas.cu:259或450网上查找N次都没找到能解决这个问题的方案，怒肝5天后终于找到了终极解决方案，供大家参考。
RuntimeError: cublas runtime error : the GPU program failed to execute at /tmp/pip–req-build-jh50bw28/aten/src/THC/THCBlas.cu:259
在搜索引擎查找N次，包括询问N次GPT导师都没找到能解决这个问题的方案，都在说cuda和pytorch的对应，但是我花大量的时间对cuda和pytorch对应好了之后，仍然出现了这个问题。怒肝5天后终于找到了终极解决方案，供大家参考。
跑某个对比实验需要用到pytorch1.1.0以及cuda10.0，通过docker pull到本地（显卡为RTX3090）一个cuda10.0的环境，然后使用torch.version.cuda()显示cuda10.0.130检验torch安装时默认的cuda版本，确保torch.version.cuda()与nvcc -V显示的版本匹配。重新配置好cuda、anaconda、python、pytorch后，运行代码，发现报错：
RuntimeError: cublas runtime error : the GPU program failed to execute at /tmp/pip-req-build-jh50bw28/aten/src/THC/THCBlas.cu:259
或者是报错：
RuntimeError: cublas runtime error : the GPU program failed to execute at /tmp/pip-req-build-jh50bw28/aten/src/THC/THCBlas.cu:450
发现torch.cuda.is_available()等等尝试均正常。即init初始化函数不会报错，但是运行到torch.bmm、torch.conv2d等实际操作会报错。
声明：本站所有文章，如无特殊说明或标注，均为本站原创发布。任何个人或组织，在未征得本站同意时，禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益，可联系我们进行处理。
