windows系统安装显卡驱动软件和CUDA11.1的详细教程

深度学习目标检测框架在进行图像计算时需要GPU进行加速,需要用到硬件GPU显卡,目标检测框架和硬件GPU建立联系需要通过①显卡驱动软件;②CUDA软件依次建立联系。这两个软件,可直接从NVIDIA官网下载,版本没有非常严格的需求,根据我们的框架要求,需让显卡驱动软件可安装的CUDA版本大于11.1,CUDA版本为11.1.

1、查询电脑硬件GPU配置(供查询合适的显卡驱动软件)

①在桌面找到“此电脑”,或进入文件夹界面的“此电脑”,右键选中“管理”,进入新界面

②在打开界面找到“设备管理器”,依次按下图操作,找到NVIDIA GPU显卡,获得显卡版本号,如下是 NVIDIA GeForce GTX 1650
windows系统安装显卡驱动软件和CUDA11.1的详细教程_第1张图片

③根据显卡型号查询驱动软件,查询网站:
https://www.nvidia.cn/Download/Find.aspx?lang=cn

windows系统安装显卡驱动软件和CUDA11.1的详细教程_第2张图片
④驱动版本和CUDA版本的对应关系:
windows系统安装显卡驱动软件和CUDA11.1的详细教程_第3张图片

⑤根据查询的配置,查询结果如下,建议下载470及以上版本驱动。
windows系统安装显卡驱动软件和CUDA11.1的详细教程_第4张图片

windows系统安装显卡驱动软件和CUDA11.1的详细教程_第5张图片

到此完成驱动软件的安装。

2、安装CUDA

①我们明确要CUDA11.1,所以直接找对应版本安装即可。
下载地址:https://developer.nvidia.com/cuda-downloads 打开进入如下界面
windows系统安装显卡驱动软件和CUDA11.1的详细教程_第6张图片

②点击进入如下界面:
<1>2-界面一:
windows系统安装显卡驱动软件和CUDA11.1的详细教程_第7张图片
<2>2-界面二
windows系统安装显卡驱动软件和CUDA11.1的详细教程_第8张图片
<3>2-界面三
windows系统安装显卡驱动软件和CUDA11.1的详细教程_第9张图片
<4>2-界面四
windows系统安装显卡驱动软件和CUDA11.1的详细教程_第10张图片
<5>2-界面五
windows系统安装显卡驱动软件和CUDA11.1的详细教程_第11张图片

③依次操作,到此即可完成CUDAToolKit的下载

3、CuDNN下载

此外,还有CuDNN需要下载安装,因为要用到深度学习库
下载网址:https://developer.nvidia.com/rdp/cudnn-download
<1>3-界面一
windows系统安装显卡驱动软件和CUDA11.1的详细教程_第12张图片
<2>3-界面二
windows系统安装显卡驱动软件和CUDA11.1的详细教程_第13张图片
<3>3-界面三
windows系统安装显卡驱动软件和CUDA11.1的详细教程_第14张图片
<4>3-界面四
windows系统安装显卡驱动软件和CUDA11.1的详细教程_第15张图片

下载解压后拷贝三个文件夹至CUDA的如下安装目录覆盖,即可完成安装
windows系统安装显卡驱动软件和CUDA11.1的详细教程_第16张图片

至此完成所有底层软件安装。

你可能感兴趣的:(windows,深度学习,人工智能,CUDA)