总结安装TensorFlow-gpu的心酸过程
1.安装前一定要查看一下各版本对应关系
https://tensorflow.google.cn/install/source_windows#gpu
2.在经过第一次因为版本不匹配的失败安装后,我选择安装1.12.0版本
3.卸载干净上一次的东西,这里可以自己去找各种卸载的教程,不多说了
4.安装python3.6,记得add to path
5.安装CUDA9.0和cudnn7.0,这里也有很多教程,一般按过程来不会出错
6.网上有很多使用Anaconda3来安装TensorFlow的教程,我第一次也是使用这个方法,但后面问题越来越多,我就干脆放弃了该方法,选择直接在cmd上使用pip安装
下面正式开始:
首先,win+R然后输入cmd,这边可以先查看一下你的python环境
输入
python
结果
python没问题了,使用pip安装指定tensorflow-gpu版本,这之前可以把pip升级到最新版
python -m pip install --upgrade pip
然后因为网络不稳定的原因,pip install tensorflow-gpu==1.12.0 时经常报 time out 的大量错误,最后找到了一种方法
指定国内镜像
pip install 包名 -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com
原文链接https://blog.csdn.net/chaokudeztt/article/details/104594127
所以在命令行中输入
pip install tensorflow-gpu==1.12.0 -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com
接着就等下载并安装成功就好啦!