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对应的文件夹就行了。

猜你喜欢

转载自blog.csdn.net/haoronge9921/article/details/85775557