CUDA和cudnn的安装

本文详细介绍了如何在Windows系统中安装和配置CUDA 10.1和TensorFlow 2.1.0。关键步骤包括:取消VisualStudioIntegration的勾选,根据当前驱动版本选择是否安装DisplayDriver,以及选择与CUDA版本匹配的cuDNN v8.0.5进行下载和安装,并将其添加到CUDA安装目录。最后,通过配置环境变量确保系统能够识别cuDNN。
摘要由CSDN通过智能技术生成

参考

  • https://blog.csdn.net/weixin_44170512/article/details/103990592
  • CUDA 和 TF的版本匹配(在最下面) : https://www.tensorflow.org/install/source_windows

准备步骤

最新版本的TensorFlow2.1.0匹配的是10.1版本的CUDA Toolkit 建议用迅雷下载

  1. CUDA下载地址 :

    https://developer.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.105_418.96_win10.exe

1.1 安装步骤

image-20210619095341879
# 注意:
1. 在CUDA选项卡下,取消勾选“Visual Studio Intergration”一项。(这是因为我们并没有使用Visual Studio开发环境)
2. 在“Driver Components”选项卡下,比较当前版本信息和安装版本信息。若当前版本高于安装版本,则取消勾选“Display Driver”;若当前版本低于或等于安装版本,保留默认信息安装即可。
3. 注意: 我的Display Driver当前版本是高于新版本的,所以不用装这个,取消勾选就行了
image-20210619095607328
## 注意:
1. 这个安装位置一般默认就好,如果你C盘不够大就修改下默认位置,然后下一步就开始安装了
image-20210619095847056 image-20210619100140331

1.2 测试是否安装成功

## 测试
1. 输入 nvcc -V 测试
image-20210619100303004

2.CUDNN 神经网络加速库

2.1 下载地址

## 所有版本下载地址 : https://developer.nvidia.com/rdp/cudnn-archive
## v8.0.5 下载地址(使用迅雷下载) : 
- https://developer.nvidia.com/compute/machine-learning/cudnn/secure/8.0.5/10.1_20201106/cudnn-10.1-windows10-x64-v8.0.5.39.zip

## 注意 
- 我们一定要选择与安装的CUDA版本相匹配的cuDNN,即选择Download cuDNN v8.0.5 (November 9th, 2020), for CUDA 10.1

2.2 安装

## 注意
- 将下载后的CUDNN压缩包解压后,将里面的cuda文件夹重命名为cudnn805然后复制到CUDA的安装目录
image-20210619102118186 image-20210619102028075

2.3 配置环境变量

## 1. 打开系统环境变量
## 2. 找到path 添加刚才添加的cudnn805文件夹下的bin目录的环境变量 然后点击上移移动到最上面
## 3. 仔细检查后点击确定保存即可
- C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\cudnn805\bin
image-20210619102921314
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

兀坐晴窗独饮茶

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值