您的位置:首页 > 其它

windows下的theano以及GPU加速环境的搭建

2016-04-15 20:09 507 查看
  笔者今年大四,是做毕设的攻坚时期,我的boss给我安排了几项任务都是简单的机器学习,像什么逻辑回归和决策树之类的。突然有一天就给我大电话说有一个师兄在研究CNN,但是现在跑起来有点慢,想让我想办法使用GPU加速一下。

  笔者发现师兄写的CNN学习库是theano,我窃喜,因为theano目前是支持GPU运行的。所以我马上答应了下来。

  后来发现在win下的环境相当难配置,所以笔者打算买个mac但是目前需要用到GPU所以勉为其难的使用我的老年机。

好吧我们进入正题。

环境

硬件:电脑 Dell 灵越 14r 5420

   CPU i5-3210

   GPU gt630M

软件:操作系统 win 7 64bit 旗舰版

   python 2.7

   CUDA 7.5

   VS  2010

环境搭建开始

VS的安装
一般来说,作为一个软件学院的学生,vs应该是人手一个的,但是我们知道,有很多同学是不喜欢使用vs的,因为这个产品冗余的功能有很多,但是我们今天要做的这个利用CUDA进行加速必须使用到vs,所以我们选择vs2010作为我们的编译环境,这里要注意的是VS2015是不支持CUDA的,至少目前是这样的,我们安装的vs最好是默认功能的,最好不要自定义安装,因为缺少某个组件的话会出现各种各样的问题的。这里的VS安装我们就不细细的讲了,这还不会的直接百度就行了。装好以后我们就进行第二项。

CUDA的安装
在装CUDA的时候,要先看看,自己的电脑支持CUDA开发吗,一般来说,NVIDIA家的卡都是可以的,但是别人家的卡就不行了。这里我们选择去官网上下载 CUDA toolkits。这里我选择的是CUDA7.5,大家可以根据自己的系统做出选择,我这里是win 7 64bit的。CUDA的安装,直接跟着提示傻瓜式安装就行了,这里建议装到C盘中,装完以后环境变量就会自动装好。

PYTHON环境的安装
我们装好CUDA的开发环境以后就开始进入到我们PYTHON环境的安装,我们都知道,PYTHON语言具有强大的库,PYTHON环境一般是建议2.7的,因为这里3.x语法并没有向下兼容。 windows系统建议直接下载打包文件 winpython,这是集成的python环境,解压直接可以用的。自带spyder编译器。并且将具有python.exe的文件目录添加到环境变量中。

theano的安装,这里我们直接去github上下载最新的源文件,地址在theano,直接下载,会有一个 压缩文件,直接解压缩,然后用dos进入这个文件夹,在命令行里面输入 :python setup.py install就可以自动将theano库安装到你自己的python中。

配置文件.theanorc.txt
在你的用户目录中(这个可以查看你的dos命令行,里面初始的路径就是你的用户目录),新建.theanorc.txt文件,并在里面写入如下内容
[global]

openmp = False

device = gpu

floatX = float32

[blas]

ldflags =

[gcc]

cxxflags = -IC:\MinGW\include

[nvcc]

flags = -LF:\python\WinPython-64bit-2.7.10.3\WinPython-64bit-2.7.10.3\python-2.7.10.amd64\libs #这里对应你的PYTHON文件中的libs

compiler_bindir = E:\Program Files (x86)\Microsoft Visual Studio 10.0\VC\bin#这里对应你的vs中的相应目录中的bin

测试
打开命令行,输入python(如果不成功,你的python路径没有放在环境变量中),然后输入import theano,只要出现using gpu的字样(这里不用在乎后面括号里出现的内容,如果出现的话),说明你的文件配置完成这样你就可以尽情享受GPU加速的快感了。

到此为止我们的环境配置就到此为止了。要是你没有完成的话,欢迎留言我会第一时间回复的。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息