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下载安装

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

3.安装tensorflow

Pip install tensorflow-gpu==1.8.0

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

猜你喜欢

转载自www.cnblogs.com/zenan/p/9328324.html