Linux下安装配置Tensorflow(GPU版本)

目录
  • 一、安装NVIDIA驱动
  • 二、gcc降级
  • 三、安装cuda
  • 四、安装cudnn
  • 五、安装Anaconda
  • 六、安装Tensorflow
  • 七、配置IPython(使用ipython/ ipython notebook作为python编辑器,适合以命令行模式运行python)
  • 八、安装PyChram(使用PyCharm作为python编辑器,适合编写python脚本)
一、安装NVIDIA驱动
1.禁用nouveau:

打开blacklist.conf:

$ sudo gedit /etc/modprobe.d/blacklist.conf

在blacklist.conf文件末尾加上这两行,保存:

blacklist nouveau
options nouveau modeset=0

应用更改:

$ sudo update-initramfs -u

重启系统,验证是否禁用nouveau:

$ lsmod | grep nouveau

没有信息返回说明已禁用


2.安装NVIDIA驱动:

方法一:
打开系统设置->软件和更新->附加驱动->选择NVIDIA驱动->应用更改
方法二:
下载nvidia驱动: http://www.nvidia.cn/Download/index.aspx?lang=cn
进入命令行界面Ctrl+Alt+F1 (深坑:笔记本的F键有小键盘,需加上Fn)
关闭图形界面

$ sudo service lightdm stop 

卸载旧版本nvidia驱动

$ sudo apt-get remove --purge nvidia* 

安装nvidia驱动

$ sudo ./NVIDIA-Linux-x86_64-361.45.11.run 

打开图形界面

$ sudo service lightdm start
3.查看NVIDIA版本检验是否安装完成:
$ nvidia-smi
二、gcc降级
1.查看编译器版本:
$ gcc --version
2.gcc版本5.0以上则降级:

tensorflow1.3只支持cuda8.0以下版本,cuda8.0不支持5.0以上的gcc编译器,因此先要降级

$ sudo apt-get install g++-4.9
$ sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.9 20 
$ sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-5 10 
$ sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-4.9 20 
$ sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-5 10 
$ sudo update-alternatives --install /usr/bin/cc cc /usr/bin/gcc 30 
$ sudo update-alternatives --set cc /usr/bin/gcc  $ sudo update-alternatives --install /usr/bin/c++ c++ /usr/bin/g++ 30 
$ sudo update-alternatives --set c++ /usr/bin/g++  
3.检查是否编译器是否已降到4.9:
$ gcc --version
三、安装cuda
1.下载cuda8.0:

https://developer.nvidia.com/cuda-80-ga2-download-archive
Installer type 选runfile (其它都安装不成功)

2.安装cuda:
$ sh cuda_8.0.61_375.26_linux.run

一直按回车

3.添加环境变量:

打开.bashrc

$ gedit ./.bashrc

在.bashrc文件末尾加上这两行,保存

export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64\${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}$ 

使环境变量生效:

$ source ~/.bashrc
$ sudo ldconfig
4.查看版本检查是否安装完成:
$ nvcc --version
5.运行cuda测试样例:

打开样例文件夹其中一个sample,运行测试样例

$ cd NVIDIA_CUDA-8.0_Samples/1_Utilities/deviceQuery
$ sudo make
$ sudo ./deviceQuery
四、安装cudnn
1.下载cuda版本对应的cudnn压缩包:

https://developer.nvidia.com/cudnn

2.解压cudnn:
$ tar -xvf cudnn-8.0-linux-x64-v5.1.tgz
3.把相应的文件拷贝到cuda文件夹:
$ sudo cp cuda/include/cudnn.h /usr/local/cuda/include/  
$ sudo cp -a cuda/lib64/libcudnn* /usr/local/cuda/lib64/ 
$ sudo chmod a+r /usr/local/cuda/include/cudnn.h 
$ sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
五、安装Anaconda
1.下载Anaconda:

https://www.anaconda.com/download/#linux

2.安装Anaconda:
$ sudo bash Anaconda3-4.2.0-Linux-x86_64.sh
3.添加环境变量:

打开/.bashrc:

$ gedit ./.bashrc

在.bashrc文件末尾加上anaconda路径,保存:

export PATH=~/anaconda3/bin:$PATH

使环境变量生效:

$ source ~/.bashrc
4.查看版本检验是否安装完成:
conda --version
python --version
pip --version
六、安装Tensorflow
1.在Anaconda创建python环境:

(格式:conda create -n evnv_name python=python_version)

$ conda create -n py35 python=3.5

激活环境:

$ source activate py35
2.选择tensorflow版本:

清华镜像: https://mirrors.tuna.tsinghua.edu.cn/help/tensorflow/
复制pip命令

3.用pip安装tensorflow:
(py35) $ pip install \  -i https://pypi.tuna.tsinghua.edu.cn/simple/ \  [https://mirrors.tuna.tsinghua.edu.cn/tensorflow/linux/gpu/tensorflow_gpu-1.3.0-cp35-cp35m-linux_x86_64.whl](https://mirrors.tuna.tsinghua.edu.cn/tensorflow/linux/gpu/tensorflow_gpu-1.3.0-cp35-cp35m-linux_x86_64.whl)
4.检验是否安装成功:
(py35) $ python
>>> import tensorflow
#退出环境:
>>> quit()
$ source deactivate py35  
5.安装keras,sklearn等库

查看环境中包含哪些python库

$ conda list -n py35

pip安装需要的库

$ source activate py35
(py35) $ pip install keras -U --pre
(py35) $ pip install sklearn
七、配置IPython
1.在创建的python环境中安装ipython:
(py35) $ conda install ipython
2.打开ipython编辑器(命令行模式)
(py35) $ ipython

退出ipython:

In [1]: exit()
3.打开ipython note ( jupyter)
(py35) $ ipython notebook

执行命令后ipython notebook以浏览器new tab的形式打开,即可以在浏览器中编辑python:


退出ipython notebook:在终端中输入ctrl+c

八、安装PyChram
1.安装jdk
2.下载PyChram:

http://www.jetbrains.com/pycharm/download/ 选community linux版

3.安装PyChram:

解压:

$ tar xfz pycharm-community-2017.3.3.tar.gz 

运行sh文件:

$ sh pycharm-community-2017.3.3/bin/pycharm.sh

安装完成

4.把PyCharm添加到桌面:

Configure->Create Desktop Entry


5.把conda虚拟环境添加到PyCharm Create New Project -> 展开Project Tnterpreter选项->

选择Existing interpreter->点击旁边的小齿轮->Add Local...



选择Conda Environment,把python路径设置为虚拟环境中的python路径(不想每次新建project都设置环境可勾选Make available to all projects)


6.png
6. 添加cuda环境变量:

新建project测试,发现tensorflow运行失败



解决方法:
Run->Edit Configurations->Environment Variables



在Environment Variables添加cuda的路径:
参考

ubuntu下安装nvidia驱动、cuda、cudnn
深度学习主机环境配置: Ubuntu16.04+GeForce GTX 1080+TensorFlow ubuntu安装TensorFlow-GPU教程 ubuntu 下GPU版的 tensorflow / keras的环境搭建
Keras安装和配置指南(Linux)
ubuntu 14.04 处理 通过anaconda启动jupyter报错
Ubuntu安装PyCharm
ImportError: libcusolver.so.8.0: cannot open shared object file: No such file or directory

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 159,015评论 4 362
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,262评论 1 292
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 108,727评论 0 243
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 43,986评论 0 205
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,363评论 3 287
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,610评论 1 219
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,871评论 2 312
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,582评论 0 198
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,297评论 1 242
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,551评论 2 246
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,053评论 1 260
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,385评论 2 253
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,035评论 3 236
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,079评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,841评论 0 195
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,648评论 2 274
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,550评论 2 270

推荐阅读更多精彩内容