首页 > 其他分享 >cublas runtime error : the GPU program failed to execute 报错

cublas runtime error : the GPU program failed to execute 报错

时间:2023-05-03 10:13:20浏览次数:40  
标签:execute 1.0 failed pytorch 报错 GPU cublas

情景

在使用 pytorch 1.0 和 RTX 3060 Laptop 训练 GIN 的时候出现了这个错误

原因

pytorch 1.0 对应的 cuda 10.0 版本太低了,不支持新的 GPU 架构

解决

最终使用了更新版本的 pytorch

标签:execute,1.0,failed,pytorch,报错,GPU,cublas
From: https://www.cnblogs.com/shixuanliu/p/17368712.html

相关文章