tensorflow GPU版本安装及配置

经检测速度大幅度上升,不枉费我折腾了这么久,最坑的就是网上教程、书都没有写将cuda的bin加入全局变量,还是根据报错信息推出来的。

 

 

1.cuda9.0下载安装

https://developer.nvidia.com/cuda-90-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

 

 

注意点:将C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin加入环境变量,其中也根据自己的安装路径。

2.cudnn7.0.5下载安装

https://developer.nvidia.com/rdp/cudnn-archive

将(根据自己安装环境)/cuda/bin加入环境变量

 

3.安装tensorflow

Pip install tensorflow-gpu==1.8.0

4.测试新开cmd下python,import tensorflow成功即可。

 

配置:

设置session的config参数

posted @ 2018-07-18 11:41  家迪的家  阅读(1546)  评论(0编辑  收藏  举报