前言
使用GPU进行深度学习框架的训练首先要保证电脑配有独立显卡,且显卡驱动正常运行。市面上常见的显卡供应商为:英伟达(NVIDIA)和AMD,与之对应的显卡驱动程序分别为:NVIDIA驱动程序(NVIDIA GeForce驱动程序) 和 Radeon驱动程序。本文是基于英伟达显卡。
1. 查看显卡的版本型号
-
右击我的电脑 -> 管理 -> 设备管理器 -> 显示适配器。(本机配置属于远古时期,勿喷)
- NVIDIA --> 表示这是英伟达显卡
- GeForce --> 表示显卡的系列为GeForce (此外主要还有RTX和Quadro)
- GTX 1650 --> 表示 GeForce系列中的显卡型号
2. 下载适配驱动
-
在英伟达显卡驱动下载官网Official Drivers | NVIDIA,按自己电脑的配置搜索驱动程序
-
在搜索到的驱动程序中列表中,下载、安装即可
-
打开设备管理器,双击显卡显示设备运转正常,说明显卡驱动安全成功
3.查看驱动版本和CUDA版本
-
在cmd中输入命令
nvidia-smi
,显示的配置信息在后续选择pytorch版本中有用
中…(img-zTa95rYw-1717512743207)]