CUDA和cudnn安装

leocloud

已于 2022-03-27 16:00:27 修改

阅读量2.6k

点赞数

分类专栏： AI 文章标签： python anaconda stm32

于 2022-03-27 15:33:05 首次发布

本文链接：https://blog.csdn.net/u012175418/article/details/123774053

版权

5 篇文章 0 订阅

订阅专栏

在Windows上安装Paddle、Tensorflow、PyTorch之前，需要安装CUDA和cudnn。

检查计算机平台的显卡驱动版本。
在NVIDIA控制面板主页查看（例如，我的版本号是466.81）：
核查NVIDIA显卡和CUDA的版本号关系
在官网上可以查询，从下图可以看到驱动版本号为466.81对应的CUDA版本为CUDA 11.3.1 Update 1以下，因此可以选择CUDA 11.2（这是由于后边需要安装Paddle，Paddle目前支持的CUDA最高为11.2）
下载后安装。
CUDA官网下载地址（需要注册，免费下载，中间最好选择自定义安装，避免重复安装显卡驱动）
在官网下载CUDA对应的cudnn。
在官网右上角的搜索栏中搜“cudnn”，下载后解压，将解压文件夹（bin，include，lib）整体复制粘贴到CUDA安装目录（默认目录是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2，粘贴需要管理员权限）
完成安装

关注

专栏目录