Win7+CUDA8.0+VS2015+Theano0.8配置GPU加速环境

来源:互联网 发布:网络上的uv是什么意思 编辑:程序博客网 时间:2024/06/10 00:22

今天感觉至少做成了一件事,心情不错!

Nvidia官网给出的兼容性情况:

这里写图片描述

大致来说:
Theano0.8的安装通过Anaconda然后pip install theano,细节参照Theano的官方文档Installation of Theano on Windows ,只要能import theano成功就可以了(或者把文档中的一个sample跑通也可以)。

CUDA直接在Nvidia官网下载,照着官方的安装提示一步步来就行,没什么坑。 安装完后可以在cmd下看一下版本(输入nvcc -V回车即可),然后运行C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0\1_Utilities\deviceQuery 下的VS工程文件(参见此文),编译运行后应该是输出电脑的GPU设备信息等,这就说明CUDA安装成功了!

然后是调用GPU,也是按照先前给的Theano官方文档去做。主要就是新建一个.theanorc.txt(里面的内容找着网上的填就行),然后需要gcc的编译器,我是用了mingw。

我先前由于没有配好VS2015的环境,总是没办法用上GPU,后来参照一些博客如: visual studio编译器和链接器使用的环境变量 ,VC++ 19 (VS2015) 编译器系统环境变量配置,具体来说要配置成能够再cmd下直接用cl命令来编译链接.c和.cpp文件。 在此过程中有一个坑,就是配置环境时,添加的PATH、INCLUDE、LIB变量要统一成32位或者64位,我当初PATH设置成(VS安装路径)…/VC/bin (该路径下的cl.exe其实是32位的),而INCLUDE和LIB都是用的x64的,结果一直报错,后来PATH改成了(VS安装路径)…/VC/bin/x86_amd64 才成功。

最后运行Theano官网给出的GPU环境测试程序,速度比单纯使用CPU提升15倍!

0 1
原创粉丝点击