CUDA(Compute Unified Device Architecture)是NVIDIA开发的并行计算平台和编程模型,用于利用NVIDIA GPU的并行计算能力,它允许开发者使用类似于C语言的编程语言编写并行程序,利用GPU的大规模并行计算能力加速各种类型的应用程序,包括科学计算、深度学习、图形渲染等,并提供了一套丰富的API和工具,使开发者能够方便地利用GPU加速计算,包括核心的CUDA Runtime API、CUDA Driver API以及各种辅助工具和库。
--------------CUDA向下兼容--------------
正文开始:
查看显卡型号及驱动:
CUDA可以向下兼容,比如我是12.4,则表示我可以安装12.4以下版本。
(安装CUDA与CUDNN与Pytorch版本需要对应)
安装cuda:
官网:CUDA Toolkit Archive | NVIDIA Developer
选择合适的版本
本文以12.1.1为例
选择解压安装包位置(随便选)
自动进入安装界面
(提示:安装CUDA前需要安装VS)
检查系统兼容性(cuda向下兼容,尽量安装适配版本)
兼容性检查完毕即可点击下一步
我这里选择的是“自定义”,也可以选精简模式。
全选。下一步。
安装 位置建议默认
安装:
安装:
查看安装位置:
检查无误