Win10下配置深度学习框架Caffe及python接口 CUDA8.0+CUDNn5.1

硬件及GPU软件说明:

GTX1080 装 + CUDA8.0 (之前在网上看到GTX1080最好装CUDA8.0,否则会出错) + CUDNN5.1 (5.0据说也会有问题)


配置过程参考:

http://blog.csdn.net/buildgood/article/details/51123312

文中需要生成_caffe.pyd文件,但推测pyd文件是debug下动态生成的,所以windows release版本的Caffe无法生成该文件,于是在此改变了文中的方法


软件下载:

     1.CUDACUDNN推荐在官网下载,CUDNNan

     2.Caffe安装包一定要在github的MicroSoft/Caffe下下载master分支: https://github.com/Microsoft/caffe

     3.Python用Anaconda安装,我的安装位置是D:/Anaconda 官网下载,好处是不用自己再安装numpy等工具包,也无需再配置环境变量

     4.google的protobuf:dos 下输入pip install protobuf即可,如遇无法用pip安装包(如prptobuf)问题,参考:

          http://blog.csdn.net/auntielee/article/details/70232362

     5.Pydot包,同上,dos下输入pip install pydotplus


具体步骤:

     1.      解压Caffe(github下载MicroSoft/Caffe)

     2.      安装Opencv2.4.10,配置环境变量及工程属性表

     3.      Anaconda安装,主要是避免自己安装python及其各种包的时候出问题(64位电脑最好下载64位的安装包)

     4.      打开Caffe.sln,修改CommenSetting,转换成Release模式(重要)

          ①  PythonSupport处,改为true

          ②  “C:/Miniconda”修改为“D:/Anaconda”

          ③  链接-附加库目录增加路径“D:/Anaconda/libs”

     5.      生成libcaffe,注意修改项目属性“将警告视为错误”为“否”

     6.      编译caffe.c,自动下载Nuget包

     7.      生成pycaffe工程-生成成功,在caffe-master\Build\x64\Release文件夹下生成了pycaffe文件夹,将该文件夹以及其下的caffe文件夹分别复制进D:\Anaconda2\Lib\site-packages文件夹中

     8.      Dos下进入python shell,输入 import caffe成功


猜你喜欢

转载自blog.csdn.net/AuntieLee/article/details/70237913