GPU驱动、CUDA和cuDNN的版本需要相互匹配与支持,否则开发环境就会出现问题。
GPU驱动版本越高,CUDA版本可选的范围就越广泛。在开发过程中,由于特殊的要求,往往需要先确定CUDA版本,那么可以根据此表确定某个CUDA版本要求的最低GPU驱动版本号,然后在安装GPU驱动和CUDA。
一、GPU驱动的选择
下载地址:
下载类型分为Game Ready驱动程序(GRD)和studio驱动程序(SD)两种,两者都可以提供完善的功能与支持,GRD类型可以在新游戏、新补丁等发布的第一时间获得支持,但如果考虑工作的稳定性和质量,还是选择SD类型。
二、CUDA版本的选择
1. 计算机管理-设备管理器-查看GPU型号
2. 确定cuda的版本
另外,安装显卡驱动后,也可以通过 nvidia-smi命令查看适用的cuda版本
3.到官网下载合适的cuda版本号
4. cuDnn版本的选择
cuDNN版本根据CUDA版本、使用的系统进行选择即可。另外,cuDNN下载需要注册账号,登录之后方可下载。
在开发环境的配置过程中,安装顺序通常为 GPU驱动 -> CUDA -> cuDNN
在确定版本的时候,若没有特殊要求,则与安装顺序一致,若有特殊要求,如运行的代码要求必须使用某个CUDA版本,则先确定CUDA版本,再确定GPU驱动与cuDNN版本。