调用GPU进行深度学习模型训练问题检查总结
建议:有可能没有将环境添加为notebook的执行kernel。在nvidia设置->帮助->系统信息->组件,可查看。建议:在官网复制其它版本的gpu安装命令进行安装尝试。
·
1、电脑是否具有nvidia的GPU显卡。
2、安装的cuda版本是否超过显卡支持的最高版本。
在nvidia设置->帮助->系统信息->组件,可查看。


3、安装的pytorch是否为gpu版,即使在官网中复制的gpu版安装命令,安装得到的也有可能为cpu版。
建议:在官网复制其它版本的gpu安装命令进行安装尝试。
4、在打开notebook后,执行代码时,反馈连接不到笔记本电脑。
建议:有可能没有将环境添加为notebook的执行kernel

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐



所有评论(0)