PyTorch-GPU版本、Tensorflow-GPU版本配置
PyTorch-GPU版本的配置
NVIDIA驱动下载
- 查看系统是否有显卡以及显卡型号(以win10为例)
- 查看方法:
- 打开控制面板
- 找到系统与安全
- 打开系统
- 找到设备管理器双击
- 找到显示适配器查看
- 查看自己NVIDIA的驱动版本(保证在410及以上)
- Win+R打开运行对话框
- 输入cmd命令
- 在命令行输入nvidia-smi查看版本号以及支持的CUDA版本
- 版本过低,可前往官网进行升级,可手动填写NVIDIA配置后搜索驱动版本进行安装,也可以选择自动安装
NVIDIA官网