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