GPU,CUDA,cuDNN,tensorflow,pytorch之间的关系

如下图所示:

GPU,CUDA,cuDNN,tensorflow,pytorch之间的关系_第1张图片

概念区分

驱动(Driver) cuda版本

指的是直接驱动GPU硬件的“驱动”模块的版本,具体来说,包含两个相互关联的内容:

1. Driver Version (形如 517.00)

该值表明驱动器的版本号,它限制了我们最高能下载哪个版本的cuda

如网站CUDA 12.1 Update 1 Release Notes 中提供的下表所示,箭头所指的就是该电脑的驱动器版本号(517.00)所在的区间,红色圆圈所值的就是最高可以支持的cuda版本(11.8.x)

GPU,CUDA,cuDNN,tensorflow,pytorch之间的关系_第2张图片

2. CUDA Version (形如 12.2)

该值指的是该电脑最高可以支持的cuda版本,而不是当前电脑上配置的、用于深度学习的cuda版本。

(ps: 其实该电脑上已经具备了该版本的cuda,只是一般情况下无法被我们的深度学习代码调用到)

运行(Runtime) cuda版本

CUDA Runtime Version (形如11.2)

该值才是我们一般意义上的、为深度学习配置的cuda版本。因此,一般来说,对于tensorflow或者pytorch,在安装它们时需要预先配置的cuda版本指的是该值。

如最上关系图所示,通过我们配置的环境变量,深度学习的代码可以访问到我们在环境变量中指定的版本的cuda

如何查看上述cuda版本以及驱动版本

1. 查看驱动cuda版本

通过下述命令行:

C:\Users\asus>nvidia-smi

结果如下图:

GPU,CUDA,cuDNN,tensorflow,pytorch之间的关系_第3张图片​​​​​

 2. 查看运行cuda版本

通过下述命令行:

C:\Users\asus>nvcc -V

结果如下图:

附:如何升级GPU驱动

 见网址:Official Drivers | NVIDIA

你可能感兴趣的:(tensorflow,pytorch,深度学习,gpu算力)