http://tensorfly.cn/tfdoc/get_started/os_setup.html
https://blog.csdn.net/yeler082/article/details/80943040
官網給出的地址是0.5的tensorflow版本,相對比較老舊。安裝是可以用指令指定安裝版本,比如安裝1.4.0版本:
pip install tensorflow-gpu==1.4
如果是安裝GPU版本,需要根據CUDA和cudnn的版本進行選擇
安裝過程中需要下載相關內容。如果安裝完成後import時報錯,比如:
1、ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory
是因爲tensorflow的版本和CUDA、cudnn版本不對,按照上表對應選擇相應的版本。
2、ImportError: No module named protobuf或者
ImportError: cannot import name pywrap_tensorflow或者
ImportError: cannot import name descriptor
之前安裝過protobuf,和tensorflow要求的版本不對應。可以先把本機的protobuf卸載,重新安裝tensorflow,安裝過程中會下載匹配的protobuf。
pip uninstall protobuf 卸載protobuf
安裝好之後根據官網例子進行測試
$ python
>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print sess.run(hello)
Hello, TensorFlow!
>>> a = tf.constant(10)
>>> b = tf.constant(32)
>>> print sess.run(a+b)
42
>>>