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

情景

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

原因

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

解决

最终使用了更新版本的 pytorch