CUDA是英伟达开发出的GPU计算框架。英伟达是制造GPU的公司,它为了让人们更好地使用自家生产出来的GPU,所以开发出CUDA用于操作GPU。如果没有CUDA,直接操作GPU则是一件非常困难的事情。

我们买回的带英伟达GPU的新电脑,会自带GPU显卡驱动,显卡驱动的版本号制约着你将来要安装的CUDA。运行nvidia-smi命令,我们可以查看自己显卡的驱动版本:

然后,根据显卡驱动版本,到英伟达官网,查找相应的CUDA开发工具包:

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

安装完CUDA开发工具包,运行nvcc -V

需要注意的是:运行nvidia-smi命令出现的窗口中也出现了CUDA版本号,如下所示:

这是画蛇添足,反而给人们造成了很大的困惑。它指的是在当前显卡驱动的版本下,可以支持的CUDA的最大版本。很多时候,当我们第一次执行nvidia-smi的时候,电脑上还并没有安装CUDA呢,所以大家务必要知道,这个窗口出现的CUDA版本号不是你所安装的CUDA的版本号。

读者推荐

要想彻底的掌握pytorch的方方面面,推荐大家关注《PyTorch面试精华》