TensorFlow更新的是真快,对于GPU版本,现在已经不支持cudnn6.0以下了,但是,我又是一个恋旧的人=_=,懒得更新cudnn,那就只能折腾以下TensorFlow,不安装最新版本了。
正常安装,直接就是
pip install tensorflow-gpu
但是,一般默认安装的是最新版,现在是1.4。
想安装1.3,怎么办,最简单的方法是这样:
pip install tensorflow-gpu==1.2
然而过程贼慢,最后就是timeout等error,我也不知道为啥,可能是外网的原因。
最后想到了一个解决办法,更改镜像源,网上一搜其实有很多,我用了的是中科大了:
pip install tensorflow-gpu==1.2 -i https://pypi.mirrors.ustc.edu.cn/simple/
OK,一切都是那么简单,在2分钟内解决了。
继续开心之旅。。
可能遇到的问题:
在import TensorFlow之后可能遇到的error:
TypeError: __init__() got an unexpected keyword argument 'syntax'
Google了n多教程,大概意思就是说我们电脑上安装了多个版本的protobuf。
首先验证一下:
import google.protobuf
print google.protobuf.__version__
要保证版本在3.0以上。否则就卸了重装
并查看protoc版本:
protoc --version
我的输出是2.6.1,网上说版本必须3.0以上,但是我没升级版本后来也解决了.
check version of protobuf from pip:
pip show protobuf
solution
sudo pip uninstall protobuf
sudo apt-get remove python-protobuf
reinstall using pip manager as follows:
sudo pip install protobuf
reference:
here