1. 程式人生 > >Ubuntu14.04下TensorFlow的安裝及配置

Ubuntu14.04下TensorFlow的安裝及配置

在安裝了cuda7.5和更新了cudnn版本之後,就可以進行TensorFlow的安裝配置了,進入官方下載地址進行下載安裝,tensorflow的github地址:https://github.com/tensorflow/tensorflow
具體安裝只需一條指令,

sudo pip install /home/lc/Downloads/tensorflow_gpu-1.2.0rc0-cp27-none-linux_x86_64.whl

更換你自己的路徑與檔名
此時安裝路徑為
/usr/local/lib/python2.7/dist-packages/
這不是本文要介紹的重點,依然要針對我安裝過程時出現的問題,進行相關的展開。

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/ubuntu/anaconda2/lib/python2.7/site-packages/tensorflow/__init__.py", line 24, in <module>
    from tensorflow.python import *
  File "/home/ubuntu/anaconda2/lib/python2.7/site-packages/tensorflow/python/__init__.py"
, line 49, in <module> from tensorflow.python import pywrap_tensorflow File "/home/ubuntu/anaconda2/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow.py", line 52, in <module> raise ImportError(msg) ImportError: Traceback (most recent call last): File "/home/ubuntu/anaconda2/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow.py"
, line 41, in <module> from tensorflow.python.pywrap_tensorflow_internal import * File "/home/ubuntu/anaconda2/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 28, in <module> _pywrap_tensorflow_internal = swig_import_helper() File "/home/ubuntu/anaconda2/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 24, in swig_import_helper _mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description) ImportError: libcusolver.so.8.0: cannot open shared object file: No such file or directory

當我Python命令列中輸入語句

import tensorflow as tf

出現以上錯誤,查了很多別人的疑問解答,都是以改變環境變數為主,

sudo vim ~/.bashrc
export LD_LIBRARY_PATH=”$LD_LIBRARY_PATH:/usr/local/cuda/lib64”
export CUDA_HOME=/usr/local/cuda
source ~/.bashrc

但我這麼做以後依然沒有解決,
仔細分析,其實不光是環境變數路徑的問題,TensorFlow現在的安裝版本適配的應該是cuda8.0的,而我安裝的是cuda7.5版本,所以需要降級版本安裝,

sudo pip install --upgrade \ https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.10.0-cp27-none-linux_x86_64.whl

可能有些人的電腦pip安裝還會出現一些問題,在安裝之前最好執行以下語句,

sudo apt-get install python-pip  

這一條是針對的Python2版本,如果你是Python3(sudo apt-get install python3-pip )
安裝之後,執行TensorFlow的執行檢測,如果正常就說明萬事大吉了。
我的是還遇到了下面的問題,

>>> import tensorflow as tf
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/usr/local/Cellar/python/2.7.6/Frameworks/Python.framework/Versions/2.7/lib/python2.7/site-packages/tensorflow/__init__.py", line 23, in <module>
    from tensorflow.python import *
  File "/usr/local/Cellar/python/2.7.6/Frameworks/Python.framework/Versions/2.7/lib/python2.7/site-packages/tensorflow/python/__init__.py", line 98, in <module>
    from tensorflow.python.platform import test
  File "/usr/local/Cellar/python/2.7.6/Frameworks/Python.framework/Versions/2.7/lib/python2.7/site-packages/tensorflow/python/platform/test.py", line 77, in <module>
    import mock                # pylint: disable=g-import-not-at-top,unused-import
  File "/usr/local/Cellar/python/2.7.6/Frameworks/Python.framework/Versions/2.7/lib/python2.7/site-packages/mock/__init__.py", line 2, in <module>
    import mock.mock as _mock
  File "/usr/local/Cellar/python/2.7.6/Frameworks/Python.framework/Versions/2.7/lib/python2.7/site-packages/mock/mock.py", line 69, in <module>
    from pbr.version import VersionInfo
ImportError: No module named pbr.version

針對這個問題,你只需要

sudo pip install pbr funcsigs

即可修復。
以上皆是我在安裝時遇到的各種問題,通過搜尋相關資料,最後都進行了有效的解決,希望能給你有用的參考。