CUDA 安装


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.exebandwidthTest.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

【侵删】