深度学习开发环境搭建教程(Mac篇)

若是你的Mac不自带NVIDIA的独立显卡(例如15寸如下或者17年新款的Macbook。具体能够在“关于本机->系统报告->图形卡/显示器”里查看),那么你可能没法在这台Mac上使用GPU训练深度学习模型。不过这并不值得遗憾。事实上,我在本身的Macbook上(15-inch,Early 2013,NVIDIA GeForce GT 650M 1024 MB)作了一个简单的测试:在mnist数据集上训练CNN模型时,GPU模式相对于CPU模式仅节省了1/3的时间。这可能要归咎于Mac上赢弱的显卡。相比之下,我更推荐购买一台有着强劲性能的显卡的PC本(好比某些游戏本)来搭建深度学习的开发环境,或者是直接租赁AWS的Instance服务。html

下面步入正题。python

安装GPU开发环境

几乎全部的主流深度学习框架在使用GPU进行模型训练时都依赖于两个底层环境:CUDA和cuDNN。前者是一个使用GPU进行并行计算的平台,后者是一个封装了使用GPU加速神经网络计算的library。git

安装CUDA

  1. 确保你的显卡被CUDA所兼容。前往CUDA-capable GPU检查可用的显卡型号。github

  2. 接下来,安装xcode(经过App Store)和命令行工具:xcode-select --install。这一步会自动安装clang编译器,能够在命令行里输入:/usr/bin/cc --version进行验证。
    clang编译器也是安装Theano的预备环境之一。shell

  3. 前往官网下载并安装CUDA8.0(dmg安装包)。注意选择相应的平台。这一步会同时安装CUDA driver和CUDA toolkit。macos

  4. 配置环境变量。打开~/.bash_profile,输入四行命令:json

    export CUDA_ROOT=/Developer/NVIDIA/CUDA-8.0 export PATH=$CUDA_ROOT/bin${PATH:+:${PATH}} export DYLD_LIBRARY_PATH=$CUDA_ROOT/lib${DYLD_LIBRARY_PATH:+:${DYLD_LIBRARY_PATH}} export LD_LIBRARY_PATH=$CUDA_ROOT/lib:$LD_LIBRARY_PATH

    其中,环境变量CUDA_ROOT给Theano指定了CUDA的安装目录,也能够被设置为/usr/local/cuda(这两个路径下的文件均经过软连接被link到了一块儿)。后端

    配置完成后,退出编辑器,source ~/.bash_profile并执行nvcc -V验证CUDA的安装是否正常。xcode

安装cuDNN

  1. 前往官网下载cuDNN压缩包。注意选择合适的平台。我下载的是cuDNN5.1 for CUDA8.0(截止到17年4月,Theano尚不支持cuDNN6.0)。bash

  2. 解压缩下载的文件,将include目录下的*.h文件复制到$CUDA_ROOT/include目录下;将lib目录下的*.so文件复制到$CUDA_ROOT/lib目录下。

至此,完成了GPU开发环境的基本部署。

安装Theano

Theano的安装极其简单:

  1. 首先,前往Miniconda安装Conda

  2. 使用Conda安装第三方依赖(“<>”里为可选包):

shell conda install numpy scipy mkl <nose> <sphinx> <pydot-ng>

  1. 使用Conda安装Theano(若是出现网络链接异常,也可使用pip进行安装。参见官网安装指南):

    conda install theano pygpu

安装Keras

Keras是一个创建在Theano和TensorFlow基础上,封装了大量底层接口,方便研究人员快速搭建模型原型的深度学习开发框架。它提供了pip和从源码直接安装两种方式。

使用pip安装:

sudo pip install keras

从源码直接安装:

git clone https://github.com/fchollet/keras.git cd keras sudo python setup.py install

Keras默认以TensorFlow为底层计算引擎。第一次执行import keras命令后,会生成~/.keras/keras.json文件。打开这个文件,将"backend": "tensorflow"修改成"backend": "theano",即完成了后端计算引擎到Theano的切换。

至此,一个基本的深度学习开发环境已搭建完成。重启电脑以确保全部配置均已生效。

环境测试

咱们能够运行一些demo来测试环境的安装是否正常。

从Github上下载Keras的源码,进入examples子目录,输入命令:

THEANO_FLAGS=mode=FAST_RUN python mnist_cnn.py

该命令将以CPU模式在mnist数据集上训练CNN模型。

输入命令:

THEANO_FLAGS='mode=FAST_RUN,device=cuda,floatX=float32,optimizer_including=cudnn' python mnist_cnn.py

该命令将以GPU模式(并使用cudnn加速)在mnist数据集上训练CNN模型。

若是一切正常,你将看到相似下图的屏幕输出:
-c

我在本身的Macbbok Pro(15寸,13年初,GT 650M的1024MB显卡)上分别测试了CPU,GPU,GPU with cuDNN三种模式下CNN模型(mnist数据集)的训练性能,其结果以下(单位:秒):

-c

出乎意料的是,若是没有cuDNN的加速,Mac上GPU的计算性能远不如CPU!而即使使用了cuDNN的加速,模型的训练也只不过缩短了1/3的时间。所以,若是想认真地玩一玩深度学习模型,仍是建议上高性能的显卡,或者直接用AWS的GPU服务吧!

相关文章
相关标签/搜索