tensorflow-gpu版本和cuda的安装

tensorflow-gpu版本和cuda的安装

最近用到tensorflow的gpu版本，安装cuda和cudnn的过程中，遇到一些坑，这里记录一下。

首先介绍一下博主的系统和显卡：Windows 10 和RTX 2080

1、先决定你要选择的tensorflow版本

这里有各个版本的tensorflow对应的cuda版本，不过楼主在这里踩坑了，因为我装的tensorflow-gpu是1.9版本，所以看了cuda是9，可是去到NVIDIA官网cuda9有三个版本，9.0，9.1，和9.2，我直接下了一个9.2结果悲剧了，各种报错，我的经验是安装完tensorflow后，首先在Anaconda Prompt中导入一下，然后就会报错，根据错误提示，就会知道你具体需要哪个版本的cuda

2、安装完对应的cuda后导入tensorflow还是会报错

这里需要安装cudnn，也不是安装，只是下载对应版本的cudnn压缩包，解压把那些文件复制到cuda对应的文件夹就行了。