随笔分类 - 人工智能

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

摘要：情景在使用 pytorch 1.0 和 RTX 3060 Laptop 训练 GIN 的时候出现了这个错误原因 pytorch 1.0 对应的 cuda 10.0 版本太低了，不支持新的 GPU 架构解决最终使用了更新版本的 pytorch 阅读全文

posted @ 2023-05-03 09:58 Frodo1124 阅读(202) 评论(0) 推荐(0) 编辑

深度学习常见问题汇总

摘要：1. Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! 原因参与运算的两个或多个变量，有的在CPU上，有的在GPU上解决首先找到报错的行，看看计算时都用到阅读全文

posted @ 2022-12-19 00:24 Frodo1124 阅读(496) 评论(0) 推荐(0) 编辑