教你安装CUDA

1.背景

学习深度学习的话,肯定需要安装PyTorch和TensorFlow,安装这两个深度学习框架之前得安装CUDA.
CUDA是什么?
CUDA是一个并行计算平台和编程模型&＃xff0c;能够使得使用GPU进行通用计算变得简单和优雅。Nvidia官方提供的CUDA 库是一个完整的工具安装包&＃xff0c;其中提供了 Nvidia驱动程序、开发 CUDA 程序相关的开发工具包等可供安装的选项。

2.安装过程

(1)选择安装版本:
首先打开自己电脑上的NVIDIA控制面板,点击系统信息

点击"组件"

这里显示的就是本机的版本,我这里是11.1.大部分电脑都可以向下兼容,所以也可以安装版本小于11.1的CUDA

(2) 到CUDA官网,点击这里

这里我选择的是10.1版本的CUDA

(3)根据自己情况进行选择,然后点击下载

(4)下载完成时,以管理员身份运行

(5)点击OK

(6)等待安装

(7)等待ing,兼容性检查

(8)点击同意

(9)选择自定义

(10)点击下一步

(11)新建文件夹,类似这种



(12)然后选择安装位置,前两个选择CUDA1,最后一个选择CUDA2,然后点击下一步

(13)勾选之后,然后点击Next

(14)等待安装

(15)点击下一步

(16)点击关闭

(17)检验是否安装成功
win &＃043; R,输入cmd打开Windows终端
输入

nvcc -V


附录:

参考自:B站"有幸遇见的上上"的视频