前言
CUDA的安装需要两个包:CUDA Toolkit
CUDNN(算是CUDA的一个补丁,用于优化深度学习)
CUDA下载官方教程:Installation Guide Windows :: CUDA Toolkit Documentation
CUDNN下载官方教程:Installation Guide :: NVIDIA Deep Learning cuDNN Documentation
一般先安装CUDA,再安装CUDNN
CUDA安装
1、确认自己自己电脑兼容的CUDA版本
搜索栏搜“NVIDIA"并点击进入NVIDIA控制面板
点击左下角的“系统信息”
查看组件部分的信息,如图中表示兼容11.6.1的CUDA
2、下载所需的CUDA版本
题主根据显卡兼容信息选择的CUDA Toolkit 11.6.1的链接下载(点前面一半的链接)
选择“Windows”→“x86_64”→“11”→“exe_{local}”→“Download(2.5GB)”