您的位置:首页 > Web前端

安装caffe如何避坑?win7 64位 + cuda7.5 + cudnn5.0 + GT630显卡

2018-03-13 14:21 429 查看
加速分为GPU和CPU加速,因为我的显卡GT630支持GPU加速,所以这两种加速都用了。caffe接口有MATLAB和Python,这里暂时不用。大家最好先整体看一下有哪些坑再装!!!

安装前的阅读:

1.检查显卡是否支持cuda并且查看自己显卡的计算能力,比如GeForceGT630计算能力是2.1。

网址:https://developer.nvidia.com/cuda-gpus。只要这里面列出的显卡就是支持cuda的。

2.下载自己显卡的最新驱动。网址:http://www.nvidia.com/Download/index.aspx。(这一步我没有做,因为安装cuda的时候已经带驱动了。有的人说是得先安装这个显卡的最新驱动,安装cuda的时候再选择自定义安装,把cuda带的驱动去掉,有可能是版本不同吧。)

3.其次选择VS版本:Visual Studio选择2013比较好。反正都是坑,最好按步骤查好后,再找到一个成功的案例来装(先装VS,后装CUDA)。

4.选择cuda版本。Cuda的选择与你的GPU型号有关,在控制面板中找到“NVIDIA控制面板”打开–“系统信息”–“组件”,最好下载这里显示的cuda版本。而且下载cuda时记住windows和linux版本区别,别下错了。(我是直接精简安装的,有的人说安装选择自定义安装,驱动不选,因为前面已经装过了,具体为什么不知道,这也是前人的经验吧。)



5.下载cuDNN库。链接:https://developer.nvidia.com/rdp/cudnn-download。上面有对应cuda版本的说明(不过建议下载网上已经试验成功的版本)。



6.下面安装时其中一个步骤很重要,是修改参数的,如下图:



1)是否用GPU加速

2)是否用cuDNN库,及cuda版本

3)是否用Python接口

4)是否用MATLAB接口

5)GPU计算能力,(现有的计算能力参数有compute_20,sm_20; compute_30,sm_30; compute_35,sm_35; compute_50,sm_50; compute_52,sm_52)。我的显卡GT630计算能力是2.1所以一开始选的 compute_21,sm_2,编译没有成功,选择compute_20,sm_20才成功,没有弄明白什么情况。

6)cuDNN压缩包解压缩完后的路径

7.前人的一些坑,下面是链接,自己参考吧。

https://zhuanlan.zhihu.com/p/23116379

http://blog.csdn.net/wonengguwozai/article/details/78846636

还有cuda版本装错了,卸载重新安装的:

http://blog.csdn.net/shuiyuejihua/article/details/78738664
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息