检查AMD的驱动是否已经更新到最新版
显卡型号查询方式:
- 设备管理器中找到显示适配器
- 通过快捷键 Crtl+Shift+esc 查看性能
AMD官网进行驱动更新
- 选择驱动与支持
- 翻到页面最下方,选择自己的显卡型号
显卡更新完毕后,利用网上的教程即可进行PyTorch的安装
推荐文章:
网上很多版本使用的不是AMD显卡,安装的为GPU版本,CPU版本进行验证时会显示False
PyTorch安装
一定要注意CPU和GPU选择是不同的
GPU参考:
CPU辅助:
安装完毕后,在PyCharm中进行环境配置
推荐教程: