【AI实战】基础环境搭建（Ubuntu+conda+tensorflow+GPU+PyCharm）

时间 2019-11-07

标签 AI实战基础环境搭建 ubuntu+conda+tensorflow+gpu+pycharm ubuntu conda tensorflow gpu pycharm 栏目 Ubuntu 繁體版

原文原文链接

为方便平常的深度学习模型开发与测试，在本身笔记本上搭建一个深度学习的基础环境，便于学习AI使用。本人使用的笔记本配置是CPU为8代i5，显卡为GTX1060，内存为8G，基本上可知足平常的AI研究与学习。下面将介绍基础环境的搭建配置过程：python

一、安装Ubuntu 18.04

（1）安装操做系统

从Ubuntu官网上下载最新的Ubuntu 18.04 LTS版本的IOS文件，而后使用Image Writer等工具，将IOS文件写入到U盘里面。
重启电脑，在BIOS里面设置第一启动顺序为U盘，保存启动，进入到Ubuntu操做系统的安装界面，按照操做指引一步一步进行安装，安装过程比较简单，在此就再也不赘述。
注意：因为本人笔记本上的内存并不大（只有8G），为了知足后续的模型训练须要，在安装操做系统时，将交换区设置得大一些，例如8G或16G，以防模型训练时读取大量样本致使内存溢出。固然，能加内存条那就更好，土豪请随意。算法

（2）安装显卡驱动

安装Ubuntu后默认安装的是开源版本的显卡驱动，为了后续可以在使用tensorflow-gpu时能更好地发挥GPU的性能，推荐安装NVIDIA官方版本的驱动。
在Ubuntu里面，打开“软件和更新”，点击里面的“附加驱动”标签页，选择使用NVIDIA driver，而后点击“应用更改”进行官方驱动的安装，安装后重启电脑便可。

重启电脑后，只要在电脑的设备信息里面看到“图形”是显示了GTX 1060，则说明NVIDIA官方显卡驱动安装成功了。

【注意】在安装NVIDIA的官方驱动时，最初是在官网上下载Cuda、Cudnn安装文件手动进行安装，安装成功了，但却没法加载驱动。后来经过在Ubuntu的“软件和更新”安装官方驱动，才顺利解决，并且安装也方便。bash

二、安装Anaconda

因为后续在使用tensorflow建模时主要使用Python语言，所以，在这里选用了当前很是流行的Python发行版本：Anaconda。Anaconda集成了大量的科学计算包，能根据须要自动下载安装软件包和相应的依赖包（p.s.这也是比pip先进之处，pip没法管理依赖包的问题）。另外，使用Anaconda还能建立虚拟环境，这样就能很方便地在同个电脑上使用Python 2.x、Python 3.x，tensorflow-cpu、tensorflow-gpu，相互之间不受影响，很是方便。

微信

（1）下载

从Anaconda官网（https://www.continuum.io/downloads）上下载安装包，选择Linux系统，安装基于Python 3.6版本网络

（2）安装

对下载的文件授予可执行权限，而后进行安装分布式

bash Anaconda3-5.2.0-Linux-x86_64.sh

当询问是否把Anaconda的bin添加到用户的环境变量中，选择yes函数

（3）建立虚拟环境

使用conda create命令建立虚拟环境到指定路径，并指定Python版本，同时能够将须要一块儿安装的包也一块儿指定。下面建立一个名为tensorflow的虚拟环境，指定使用Python 3.6版本，并在虚拟环境中安装numpy、scipy、matplotlib、jupyter等软件包，命令以下：工具

conda create –n tensorflow python=3.6 numpy scipy matplotlib jupyter

其中-n指定虚拟环境的名称
默认安装的路径位于anaconda安装目录下的envs文件夹里面，也可使用—prefix参数来从新指定虚拟环境路径oop

若是要建立第二个虚拟环境，只需使用不一样的名称便可，若是建立一个名为tensorflow2，指定使用Python 2.7版本，则命令以下：性能

conda create –n tensorflow2 python=2.7

若是要查看有哪些虚拟环境，则执行如下命令：

conda info -envis

【注意】若是在建立conda虚拟环境时没有指定python的版本，则默认是使用anaconda安装目录下bin中的python版本。为了实现虚拟环境的隔离，必须指定python版本

（4）配置conda国内镜像源

为了提高conda的安装软件包时的下载速度，建议对conda配置国内的镜像源（清华大学conda镜像源），配置方式以下：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

若是要切回默认源，则执行

conda config –remove-key channels

【注意】网上不少配置conda国内镜像源的教程都只写着只配置清华大学conda的free镜像（https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/），但该源的不少软件包并非最新版本，为了能更新最新版本，再指定一个清华大学conda的main镜像（https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/），这样就能安装、更新最新版本的软件了。

（5）激活虚拟环境

建立好conda虚拟环境后，在使用以前必须先进行激活。下面激活刚建立的tensorflow虚拟环境，命令以下：

conda source activate tensorflow

若是要注销退出当前的虚拟环境，则执行命令：

conda source deactivate tensorflow

很是方便

三、安装tensorflow-gpu

为了提高AI模型的训练效率，安装使用tensorflow-gpu进行模型训练。
激活进入刚才建立的tensorflow的conda虚拟环境

conda source activate tensorflow

而后使用如下命令安装tensorflow-gpu

conda install tensorflow-gpu

conda将会检测tensorflow-gpu的最新版本以及相关的依赖包，包括调用NVIDIA显卡所须要的Cuda、Cudnn等依赖环境，都会自动按顺序进行安装，很是方便吧。
若是须要升级tensorflow-gpu的版本，则执行如下命令进行更新

conda upgrade tensorflow-gpu

注意：并非全部软件包都能直接安装，有一些须要指定安装源，例如作AI中常常会用到的OpenCV即是没法直接安装的，须要指定安装源，命令以下：

conda install --channel https://conda.anaconda.org/menpo opencv3

四、安装PyCharm

PyCharm是一个优秀的Python开发IDE环境，特别是还提供了社区版能够无偿使用，业界良心啊！

（1）安装PyCharm

在Ubuntu里面安装PyCharm很是简单，在Ubuntu软件商城里面搜索“pycharm”，而后选择社区版“PyCharm CE”进行安装便可，很是方便。

（2）在PyCharm中使用conda虚拟环境

为了可以在PyCharm中使用咱们本身建立的conda虚拟环境，须要进行下配置。在Pycharm的Files>>settings>>Project Interpreter>>Add local 里面添加刚才建立的conda虚拟环境的目录下所在的Python 3.6程序，应用以后就可使用咱们本身使用的虚拟环境了，以下图所示：

至此，“AI实战”的基础环境搭建就已所有配置完成了。

后面将陆续推出更多AI实战内容，敬请留意。

推荐相关阅读

欢迎关注本人的微信公众号“大数据与人工智能Lab”（BigdataAILab），获取更多信息