1、查看电脑是否支持GPU加速
要想实现GPU加速,必须安装cuda,电脑拥有NVIDA独立显卡才能支持cuda的安装,AMD显卡不支持cuda。具体查看步骤如下:
- 打开设备管理器,找到显示适配器,查看是否有NVIDA显卡:
- 打开NVIDA官网(地址:https://developer.nvidia.com/cuda-gpus)查看显卡型号是否在NVIDA列表,存在则可以安装cuda实现GPU加速。
2、CUDA的下载安装
- 安装前准备
- 查看是否有NVIDA独立显卡,
- 查看是否有NVIDA显卡驱动程序,没有则下载安装,下载地址:https://www.nvidia.cn/geforce/drivers/
安装完成可查看GPU显卡所支持的CUDA版本:
开始->NVIDIA控制面板->帮助->系统信息->组件->nvidia.dll后边cuda参数,下图显卡支持10.2版本:
-
- 查看是否有Visual Studio
- 下载CUDA:
下载地址:https://developer.nvidia.com/cuda-toolkit-archive,选择合适的版本下载即可。
- 下载完成安装包,点击“下一步”等选项进行安装。安装完成后,系统中增加了CUDA_PATH和CUDA_V9_0两个环境变量。
- 查看是否安装情况:
- 查看版本:打开cmd命令框,输入nvcc --version
- 查看变量情况:输入set cuda
- 验证deviceQuery和bandwidthTest:命令窗口,定位到cuda安装目录:"C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\extras\demo_suite",分别输入deviceQuery.exe和bandwidthTest.exe运行,当Result=PASS说明通过,反之,Rsult=Fail 则需要重新安装。
- 安装结束:如果以上检测没有问题,说明cuda安装成功。
可以下载CUDNN更快加速。CUDNN是一个常见的神经网络层加速库文件,能够很大程度上把加载到显卡上的网络层数据进行优化计算。CUDA加速主要依靠显卡,CUDNN在有CUDA的基础上运行,速度可在CUDA基础上加速2倍以上。
补充:
- nvidia-smi:这里看到的CUDA Version版本是指可驱动的最高版本,下载的CUDA版本应小于等于此版本;
- nvidia控制面板:这里看到的版本依旧指可驱动的最高版本;
- nvcc --version:这里输出的版本即为CUDA运行版本。
参考:
码农的后花园的文章 - 知乎 https://zhuanlan.zhihu.com/p/143881055
码农的后花园的文章 - 知乎 https://zhuanlan.zhihu.com/p/144311348
【侵删】