对应版本_GPU驱动CUDAPyTorch对应关系

GPU-驱动-CUDA-PyTorch版本之前是存在依赖的，因此在安装之前建议查询四者之间的关系，不然很容易在安装环境或者在运行的过程中出现各种奇怪的问题。

这是CVHub公众号的第四篇原创文章，也是《学术小白也能看懂的学术进阶专栏》(计算机视觉方向)的第四篇文章！

查询显卡能支持的最新驱动

在 NVIDIA 官网能够找到自己的显卡所对应的最新驱动的版本。

https://www.nvidia.cn/Download/index.aspx?lang=cn# 

以 GTX 1050 为例：

1. 首先在第一个候选框中选择对应的产品线，GTX1050 属于 GeForce 产品线

2. 然后在产品系列里面选择 10 系列，如果你的是笔记本，记得选 Notebooks

3. 然后在产品家族中选择 1050

4. 然后选好自己对应的操作系统即可，剩下两个选项默认就好

5. 点击搜索按钮，就会跳转到以下界面

       因此我们可以确定，GTX1050显卡在Win10的最新驱动版本为451.48

查询驱动能够支持的 CUDA 版本

从上面的步骤我们可以看到 GTX1050 支持的最新驱动版本为 451.48, 然后我们可以在 CUDA文档中查看该驱动版本能够支持的 CUDA 版本

https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

       

因为 GTX1050 最新驱动为 451.48，因此小于等于此版本的驱动的 CUDA 都能安装。如上图所示，1050ti 最新的驱动能够支持 11.0.2GA 及以下所有版本的 CUDA。

查询 Pytorch 支持的 CUDA 版本

在 pytorch的官网我们可以查询到 pytorch 不同版本能够支持的 CUDA 版本。

https://pytorch.org/get-started/previous-versions/

以 pytorch1.5.0 为例：

       

其支持的 CUDA 版本有：9.2，10.1，10.2

交流渠道

微信群(长按加入)：