装驱动
一般电脑都会装好驱动
windows中搜索NVIDIA控制面板再点击下图键即可也可看到驱动版本。
命令行nvidia-smi命令
也可以查看驱动版本,其中显示的cuda是最高支持版本
装cuda
11.8版本
自定义安装步骤中,已经装好驱动就别勾选了。
nvcc --version
命令检验是否装好,可查看当前cuda版本。
困惑:对于11.8的cuda,下载时还有个Version选项不知道啥意思?
环境变量只需要加这个就行
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8
nvcc是NVIDIA CUDA Compiler,可用于编译cuda程序。
在CUDA中,host和device是两个重要的概念,我们用host指代CPU及其内存,而用device指代GPU及其显存。CUDA程序中既包含host程序,又包含device程序,它们分别在CPU和GPU上运行。
装cuDNN
cuDNN全称NVIDIA CUDA® Deep Neural Network library, 是一个用于深度神经网络的GPU加速库。cuDNN包含了为神经网络中常见的计算任务提供高度优化的实现。包括前向卷积、反向卷积、注意力机制、矩阵乘法(matmul)、池化(pooling)和归一化(normalization)等。
cuDNN当下好像要注册英伟达账号才能下载,下载链接点击去会提示照着流程来即可。
cuDNN历史版本下载地址
把解压后的对应文件新复制到cuda安装对应名字的文件夹下即可