環境配置-Linux下安裝tensorflow

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
>>>
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章