深度学习目标检测框架在进行图像计算时需要GPU进行加速,需要用到硬件GPU显卡,目标检测框架和硬件GPU建立联系需要通过①显卡驱动软件;②CUDA软件依次建立联系。这两个软件,可直接从NVIDIA官网下载,版本没有非常严格的需求,根据我们的框架要求,需让显卡驱动软件可安装的CUDA版本大于11.1,CUDA版本为11.1.
1、查询电脑硬件GPU配置(供查询合适的显卡驱动软件)
①在桌面找到“此电脑”,或进入文件夹界面的“此电脑”,右键选中“管理”,进入新界面
②在打开界面找到“设备管理器”,依次按下图操作,找到NVIDIA GPU显卡,获得显卡版本号,如下是 NVIDIA GeForce GTX 1650
③根据显卡型号查询驱动软件,查询网站:
https://www.nvidia.cn/Download/Find.aspx?lang=cn
④驱动版本和CUDA版本的对应关系:
⑤根据查询的配置,查询结果如下,建议下载470及以上版本驱动。
到此完成驱动软件的安装。
2、安装CUDA
①我们明确要CUDA11.1,所以直接找对应版本安装即可。
下载地址:https://developer.nvidia.com/cuda-downloads 打开进入如下界面
②点击进入如下界面:
<1>2-界面一:
<2>2-界面二
<3>2-界面三
<4>2-界面四
<5>2-界面五
③依次操作,到此即可完成CUDAToolKit的下载
3、CuDNN下载
此外,还有CuDNN需要下载安装,因为要用到深度学习库
下载网址:https://developer.nvidia.com/rdp/cudnn-download
<1>3-界面一
<2>3-界面二
<3>3-界面三
<4>3-界面四
下载解压后拷贝三个文件夹至CUDA的如下安装目录覆盖,即可完成安装
至此完成所有底层软件安装。