为了方便调试,在Windows上安装了一套CUDA和CuDNN,记录一下过程。
1. 安装显卡驱动(已安装驱动可跳过)
在一台配有GPU的电脑上安装CUDA,首先要确保电脑已经安装了显卡驱动。若未安装,可在NVIDIA官网下载并安装对应的驱动,驱动搜索界面如下:
2. 安装CUDA
CUDA是NVIDIA研发的并行计算平台和变成模型,可利用GPU的能力大幅提升运算效能。
在安装CUDA之前,需要查看自己电脑适配的驱动版本。命令行执行nvidia-smi,查看显卡信息和CUDA版本:
可以看到,目前适配的CUDA驱动版本是12.0。我们可以安装12.0或更早的CUDA runtime版本,我选择的版本是11.7。
CUDA安装文件下载地址: