GPU驱动、CUDA和cuDNN之间的版本匹配与下载

GPU驱动、CUDA和cuDNN的版本需要相互匹配与支持,否则开发环境就会出现问题,为了更好的了解版本之间的匹配问题,可以先了解GPU、CUDA和cuDNN分别是什么,三者之间又是什么样的关系。可参考这篇文章GPU、CUDA和cuDNN分别是什么,之间又有什么关系?

1. GPU驱动

GPU驱动程序根据自己计算机所安装的显卡型号下载
NVIDIA GPU驱动程序下载地址


按照自己的显卡型号,选择正确的产品类型、系列、家族所使用的操作系统
其中,下载类型分为Game Ready驱动程序(GRD)和studio驱动程序(SD)两种,两者都可以提供完善的功能与支持,GRD类型可以在新游戏、新补丁等发布的第一时间获得支持,但如果考虑工作的稳定性和质量,还是选择SD类型。

2. CUDA

CUDA版本则是根据自己安装的驱动程序版本来选择

2.1. 查看自己GPU驱动的版本

首先,打开NVIDIA控制面板,可以在所有应用中查找或者使用系统搜索查找;然后,可以在系统信息中看到驱动版本号



另外,在驱动程序安装之后,还可以在终端中输入nvidia-smi命令来查看GPU驱动版本


2.2. 确定与GPU驱动匹配的CUDA版本

根据NVIDIA CUDA版本对应文档来查看符合自己GPU驱动的CUDA版本


GPU驱动版本越高,CUDA版本可选的范围就越广泛
在开发过程中,由于特殊的要求,往往需要先确定CUDA版本,那么可以根据此表确定某个CUDA版本要求的最低GPU驱动版本号,然后在安装GPU驱动和CUDA

2.3. 下载CUDA

CUDA下载地址
选择适合的CUDA版本,然后根据自己使用的系统进行选择下载



3. cuDNN

cuDNN下载地址
cuDNN版本只需要根据CUDA版本进行选择即可,然后在根据使用的系统选择下载
另外,cuDNN下载需要注册账号,登录之后方可下载



在开发环境的配置过程中,安装顺序通常为 GPU驱动 -> CUDA -> cuDNN
在确定版本的时候,若没有特殊要求,则与安装顺序一致,若有特殊要求,如运行的代码要求必须使用某个CUDA版本,则先确定CUDA版本,再确定GPU驱动与cuDNN版本

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐