【AI】如何安装CUDA
希望星尘
编辑于 2023年04月13日 02:19

什么是CUDA

CUDA是将英伟达显卡GPU用于各类大规模并行计算的任务的计算平台,比如我们的AI训练就是其中一种,安装了CUDA Toolkit后就能使用GPU参与此类计算,可大大加速计算过程。

众所周知,CPU比较适合进行串行计算,拿并行计算没什么办法,效率极其低下。而GPU却生来就适合进行大规模并行计算,因此作为生产力工具之一的CUDA是必不可少的。

具体内容可参照:

CUDA Release Notes (nvidia.com)

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

如何安装CUDA

首先在命令提示行内输入`nvidia-smi查看自己的英伟达显卡的驱动中的CUDA版本号(前提是英伟达显卡,且正确安装合适的显卡驱动),然后根据该版本选择相同版本的CUDA Toolkit安装即可。比如我的RTX 4080就适合目前的12.1版本CUDA。

查看CUDA版本

CUDA Toolkit下载前提条件:

前往英伟达开发者首页(https://developer.nvidia.com/)注册账号,然后选择CUDA Toolkit下载。默认情况下就是下载最新的Toolkits,如果最新的不适合你的CUDA版本,点击下方Resources第一行第二列的Archive of Previous CUDA Releases,即可打开之前版本的CUDA下载页面,选择合适的版本下载就可以咯。

前往英伟达官网下载CUDA