Pip installation
注意:pip安装会影响其他的python程序
【下载对应版本的whl文件】
https://github.com/tensorflow/tensorflow
【安装】
sudo pip install tensorflow_gpu-0.12.1-cp27-none-linux_x86_64.whl
问题及解决方案
【ImportError: libcudart.so.8.0: cannot open shared object file: No such file or directory】
解决方法:先卸载cuda-7.5,再安装cuda-8.0
卸载步骤:
sudo rm -rf /usr/local/cuda-7.5
sudo rm -rf /usr/local/cuda
sudo apt-get autoremove(必须运行才能卸载7.5)
【ImportError: cannot import name pywrap_tensorflow】
在~/.bashrc中添加路径
export PYTHONPATH=$PYTHONPATH:/usr/local/lib/python2.7/dist-packages
【I tensorflow/stream_executor/dso_loader.cc:119] Couldn't open CUDA library libcudnn.so. LD_LIBRARY_PATH】
在~/.bashrc中添加路径
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
注意更新链接库:sudo ldconfig
【如果tensorflow版本不对,则卸载升级】
sudo pip uninstall tensorflow
Virtualenv installation
注意:虚拟环境安装不会影响系统上其他python程序
【安装pip和virtualenv】
sudo apt-get install python-pip python-dev python-virtualenv
【创建虚拟环境】
virtualenv --system-site-packages ~/tensorflow
注意:如果不想用系统的packages,则运行
unset PYTHONPATH
virtualenv --no-site-packages ~/tensorflow
【激活虚拟环境】
source ~/tensorflow/bin/activate
【安装tensorflow】
pip install tensorflow_gpu-0.12.1-cp27-none-linux_x86_64.whl
【测试】(在虚拟环境中进行测试)
(tensorflow)$ python
>> import tensorflow
【停用tensorflow】
(tensorflow)$ deactivate