1.背景
学习深度学习的话,肯定需要安装PyTorch和TensorFlow,安装这两个深度学习框架之前得安装CUDA.
CUDA是什么?
CUDA是一个并行计算平台和编程模型,能够使得使用GPU进行通用计算变得简单和优雅。Nvidia官方提供的CUDA 库是一个完整的工具安装包,其中提供了 Nvidia驱动程序、开发 CUDA 程序相关的开发工具包等可供安装的选项。
2.安装过程
(1)选择安装版本:
首先打开自己电脑上的NVIDIA控制面板,点击系统信息
点击"组件"
这里显示的就是本机的版本,我这里是11.1.大部分电脑都可以向下兼容,所以也可以安装版本小于11.1的CUDA
(2) 到CUDA官网,点击这里
这里我选择的是10.1版本的CUDA
(3)根据自己情况进行选择,然后点击下载
(4)下载完成时,以管理员身份运行
nvcc -V
附录:
参考自:B站"有幸遇见的上上"的视频
文章出处登录后可见!
已经登录?立即刷新