Windows下安装CUDA+CUDNN并使用tensorflow-gpu

一、前言

　　CUDA和CUDNN要使用对应的版本，我使用的是CUDA11.0+CUDNN8.1.1

二、CUDA下载安装

官网下载：https://developer.nvidia.com/cuda-11.0-download-archive

 下载exe文件后点击安装会进行解压，解压路径建议默认，解压完成后会自动删除。

 

 

 

如果没有下载VS就取消Visual Studio Integration，否则会出现安装错误 

 

选择安装路径并记下来,C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0 ，后面配置环境会用到。

 

 

三、CUDNN下载配置

官网下载：https://developer.nvidia.com/rdp/cudnn-archive

官网下载需要注册，注册可能会有困难，可以用分享链接

链接：https://pan.baidu.com/s/1WrcKAm3PfermGg55WGwVUA?pwd=6r94
提取码：6r94

下载好后解压，获取bin，include，lib三个文件夹，将三个文件夹复制到CUDA的安装路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0下并进行替换

 四、环境配置

在环境变量->系统变量->Path中添加C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\include;C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\lib

 配置好后cmd输入 nvcc -V 可以查看CUDA是否安装成功

四、tensorflow-gpu使用

如果安装了CPU版的先将CPU版卸载，再使用命令下载GPU版，pip install tensorflow-gpu==2.5.0

import tensorflow后运行pycharm可以看到一串successful与显卡信息，代表tensorflow-gpu安装成功了