ubuntu上安装tensorflow-gpu(前提是gpu驱动都配置好)

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

                  https://blog.csdn.net/wangguchao/article/details/81776177 

                  https://www.cnblogs.com/yasmi/p/4832461.html

发布了38 篇原创文章 · 获赞 8 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/u014090429/article/details/91430926
今日推荐