win 10+tensorflow-gpu+cuda+cudnn

装了三天的tensorflow-gpu,看了无数的博客,试了无数次方法,踩了无数的坑,今天终于给他喵的整好了。写个博客给大家填个坑。

首先查一下显卡配置:
在命令行中输入nvidia-smi这里可能需要配置环境变量C:\Program Files\NVIDIA Corporation\NVSMI
找到Driver version版本
然后去官网找到对应关系:
对应关系
由于我的是399.24所以我装的cuda版本应该是9.2.148
再去找对应tensorflow的版本win10下的
看以看出我可以装1.6到1.13,cudnn的版本为7.
OK,准备工作做完了。现在去下载anacondawin10 64bit按照步骤一步一步来就能安好,这挺简单的。但是要注意要选上Add … PATH这样就不用添加环境变量了。
在装完后可以改一下conda的下载源。这个可以参考我以前写的博客换清华源
现在导包:
命令行中创建虚拟环境conda create -n tensorflow-gpu python=3.6这里建议装3.6。
然后激活环境activate tensorflow-gpu就是进入刚才间的文件夹(tensorflow-gpu)
导入tensorflow包conda install tensorflow-gpu=1.13版本根据自己的选择
在这里个地方可能会出错要导入的包
可以看出这两个并不适合我的版本,别急先导里面。
下载完成后conda install cudatoolkit=9.2 conda install cudnn=7
然后去nivdia官网下载cuda和cudnncudatoolkit
cudnn
CUDA位置默认就好方便下面的操作,cudnn下载需要注册。
下载完成后将cudnn文件解压分别将文件夹下面的文件copy到CUDA对应的文件夹下。
到这个地方应该完成一大半了。这样就可以在命令行中运行了。(要先激活环境,也就是进入tensorflow-gpu文件夹)
想要在pycharm中运行的话还需要在pycharm中进行配置
首先需要 project interpreter 这个很简单就是找到文件夹位置这个文件夹下的python.exe
然后就是将这个弄好Run --> Edit Configurations
配置 Edit Configurations
现在就都弄好了(我的是这样)。

发布了49 篇原创文章 · 获赞 18 · 访问量 1433

猜你喜欢

转载自blog.csdn.net/qq_44099721/article/details/102572851