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

发布时间 2023-05-03 10:03:47作者: Frodo1124

情景

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

原因

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

解决

最终使用了更新版本的 pytorch