ubuntu上安装tensorflow-gpu(前提是gpu驱动都配置好)
release版本安装
依赖环境:python
准备工作:
(1). 首先更新自己python版本为3.0+
安装python3
apt-get install python3
安装pip3
apt-get install python3-pip
为python3添加包
pip3 install packagename
安装pillow
apt-get install libjpeg-dev libfreetype6-dev zlib1g-dev libpng12-dev
pip3 install pillow
(2). 将python3设置为默认版本
sudo update-alternatives --install /usr/bin/python python /usr/bin/python2 100
sudo update-alternatives --install /usr/bin/python python /usr/bin/python3 150
正式安装:
(1). 安装pip:
sudo apt-get install python-pip python-dev # for Python 2.7
sudo apt-get install python3-pip python3-dev # for Python 3.n
(2). 利用pip安装Tensorflow:
sudo pip install tensorflow-gpu
测试:
python
import tensorflow as tf
可能遇到的错误:
ImportError: libcublas.so.10.0: cannot open shared object file: No such file or directory
这是tensorflow版本和cuda版本不匹配导致的,具体可查表格
解决方案:
如果你的cuda是8.0,那么你可以使用下面的指令将tensorflow的版本回滚到1.4版本
pip install --upgrade tensorflow-gpu==1.4
参考链接:https://blog.csdn.net/u014797226/article/details/80229887
https://www.jianshu.com/p/6f54b4f96556