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参数


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM