模型训练首先小白需要安装GPU版的pytorch 便于自己方便也为了给后面的人一些指导特意写下。 第一步:安装NVIDIA的驱动程序 windows参考: https://blog.csdn.net/fengxinzioo/article/details/105646969?utm_medium=distribute.pc_relevant.none-task-blog-2%7Edefault%7EBlogCommendFromMachineLearnPai2%7Edefault-1.readhide&depth_1-utm_source=distribute.pc_relevant.none-task-blog-2%7Edefault%7EBlogCommendFromMachineLearnPai2%7Edefault-1.readhide ubuntu参考: 第一种方法:(最快,最省事,笔者较喜欢) 1、点击菜单栏 2、点击附加驱动 3、稍等过后里面会显示各种驱动