3070/3080/3090显卡 + win10 下的tensorflow2-gpu配置

准备工作:

确定gpu与显卡的对应关系

 

1、安装CUDA11.1

链接：https://pan.baidu.com/s/1yKfgpLyFeGHpGRFPbCKMRw 
提取码：1234  

安装过程保持默认

2、下载cudnn包 (执行以下步骤，无需配置环境变量)

链接：https://pan.baidu.com/s/1T1aUvxvRajgxuEGGwCMm4w 
提取码：1234 

将cudnn下的cuda中的文件夹bin下的cudnn64_8.dll复制到cuda安装目录下对应的bin下

cudnn的lib/x64下的cudnn64_8.lib 复制到cuda对应的x64下

cudnn下的include复制到cuda对应的安装目录下

3、安装tensorflow的gpu

4、安装成功测试