摘要: 如何使用GPU训练Pytorch模型 这两天的深度学习实验真实让人头疼,传说中的“猫狗大战”,对模型的训练用CPU的话9h起步,12h是常态,大学生哪耗得起,因此查找资料搭建了GPU环境。 确定CUDA版本 首先你需要知道自己的电脑处理器,再去官网查看对应的CUDA版本: 我的处理器low了点,只配 阅读全文
posted @ 2022-05-02 21:10 摸鱼小天才 阅读(258) 评论(0) 推荐(0) 编辑