【如有错误欢迎指正!!!】
1.前言
本文重点在于安装tensorflow,鉴于在安装双系统还有很多小白,我在这把详细的图文教学贴上。目前在研究深度学习的东西,刚入门,想先拿自己的破电脑搭建一个gpu版本的tensorflow用于跑相对简单点的网络。博主的电脑配置很垃圾的,是自己很久前组装的电脑。GPU:Nvidia Geforce 760M·2G;8G内存。本文化包含了Ubuntu系统启动盘的制作,Ubuntu系统的安装,tensorflow-gpu的搭建。自己在搭建tensorflow的时候遇见过很多坑,也花费了很多时间解决,希望把自己的经验写下来,能够帮助更多的人,减少大家在搭建tensorflow的时候花费的时间。tensorflow-gpu版本要求GPU的计算能力在3.0以上,请大家参考【Nvidia GPU运算能力列表】一文。
2.准备工作
2.1 安装Ubuntu16.04
2.1.1 制作启动盘
从Ubuntu官网上下载一个镜像(必须是64位,目前tensorflow只支持64位系统),随便找一个4G以上的优盘自己使用UltraISO做一个系统盘就行。步骤如下:
(1)使用UltraISO打开镜像系统,如图1
(2)从菜单栏里 启动,如图2
(3)点击写入硬盘映像,在出现的画面中(如图3)写入方式选择 USB-HDD+,然后点击写入,你会看到消息中有“开始写入...”,现在我只需要静静的等待。
(4)当刻录完成时,你会看到消息中的“刻录成功!”。然点击退出即可。如图4。到这一步系统盘制作成功。
2.1.2 系统安装
(1)在Windows下给自己腾出一个60G左右的空盘,然后右击桌面上的‘我的电脑-->‘管理-->存储-->磁盘管理’,如图5,找到自己的空盘符,在空盘符上右击,点击删除卷。
(3)四个选项中你可以选择(a)或者(b)。(a)和(b)的区别在于点击(a)我们可以在不安装Ubuntu的情况下体验该系统,而(b)是直接进行安装。下边我们看看一下选(a)的画面 如图6,选择(b)的画面如图7。我在这选择(a)。
(4)如果我们不着急安装系统,我们可以先体验一下Ubuntu的简版。如果想直接安装,我们点击图6桌面上的“Install Ubuntu 16.04.3 LTS”进行安装。然后会出现图7的画面,这个是和(1)中直接点击(b)出现的画面是一模一样的,在图7中我们选择‘中文简体’,如果你想体验英文的话,你可以选择‘English’。点击继续。画面如图8所示。我们什么都不选,直接点击‘继续’。
(5)我们接下来要进行分区,这里的空闲(如图9)就是我们刚才从Windows下删除的卷,接下来我们要将这个空闲进行分区。
(6)在分区界面我们点击空闲内存,然后点击左下角‘+’号,第一次我们添加根分区‘/boot’(200M 左右呢,我分了256M),此分区Linux的内核及引导系统程序所需要的文件,比如 vmlinuz initrd.img文件都位于这个目录中。在一般情况下,GRUB或LILO系统引导管理器也位于这个目录;启动撞在文件存放位置,如kernels,initrd,grub。如图10 选择Ext4 日志文件系统,挂载点/boot。下一步我们将处理其他分区。
(7)接下来我把完整的分区放在下表中(参考博客链接)(挂载点以及类型要选对):注意。。如果分区某个分区分错了,选中该分区,点击左下角‘-’号。
创建完分区后如图11,最下边‘安装启动引导器设备’选择/boot的分区,在这我选择/dev/sda11,然后点击现在安装。
(8)接下来会出现图12,在中国的领土上随便选。然后点击‘继续’。
(9)点击继续后会出现图13,到这里我们只需要静静的等待即可。等安装完成,我们可以重启电脑了。
3.搭建tensorflow-gpu(敲黑板!!!)
3.1 下载Nvidia驱动
3.2 cuda8.0和cudnn v5.1下载(1.87G)及安装
(1) cuda8.0和cudnn v5.1下载
(2)cuda8.0安装
cd 下载/ sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb sudo apt-get update sudo apt-get install cuda
(3)降低gcc版本,cuda8.0不支持gcc5.0,我们降低到4.9
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++
在终端中输入以下命令(注意空格键)
tar xvzf cudnn-8.0-linux-x64-v5.1.tgz sudo cp cuda/include/cudnn.h /usr/local/cuda/include sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64 sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn* sudo find /usr/ -name 'libcudnn.so'(找到libcudnn.so的路径,我的路径是/usr/local/cuda/targets/x86-64-linux/lib)
sudo cp /usr/local/cuda/targets/x86-64-linux/lib/libcudnn.so /usr/lib
sudo find /usr/ -name 'libcufft.so'(我的位置是/usr/lib,如果和我的路径一样就不需要复制了,否则和上边一样,把找到的libcufft.so复制到/usr/lib中)
(5)配置环境变量
sudo gedit ~/.bashrc(在最后一行加入如下两条) export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:$CUDA_HOME/lib64:$CUDA_HOME/extras/CUPTI/lib64" export CUDA_HOME=/usr/local/cuda source ~/.bashrc
(6)安装tensorflow-gpu0.12.0(联网下进行!)
sudo apt-get install python-pip python-dev sudo pip install tensorflow-gpu==0.12.0
(7)测试tensorflow-gpu,输入下面两天命令,出现下图,安装成功!
python import tensorflow as tf
到这就大功告成了!