Win10 +VS2017+ python3.66 + CUDA9.2 + cuDNNv7.2.1 +微软版pycaffe编译

最新推荐文章于 2022-03-29 12:32:49 发布

qq_41895190

最新推荐文章于 2022-03-29 12:32:49 发布

阅读量3.8k

点赞数 2

本文链接：https://blog.csdn.net/qq_41895190/article/details/82414695

版权

本文档详细介绍了如何在Windows 10环境下，使用Visual Studio 2017、CUDA 9.2和CuDNN v7.2.1编译PyCaffe。教程涵盖了从下载所需文件、配置CUDA和CuDNN、修改项目属性，到解决VS编译错误和设置生成事件的全过程。最后，文章提供了解决常见问题的建议，以及MNIST数据集的测试步骤。

摘要由CSDN通过智能技术生成

Win10 +VS2017+ python3.66 + CUDA9.2 + cuDNNv7.2.1 +微软版pycaffe编译

需要准备的文件:

1，VS2017，这个不再赘述

2，Windows版的caffe，微软链接为 https://github.com/Microsoft/caffe

BVLC官方链接为https://github.com/BVLC/caffe/tree/windows

安装过程：

1，将下载的caffe-windows.zip进行解压，并进入其根目录下的windows目录，（eg:D:\MS-caffe-master）。将该目录下的CommonSettings.props.example复制一个并改名为CommonSettings.props，同样还是放在当前目录下。

2，点击打开该CommonSettings.props，这里由于本人的需要，对matcaffe，和pycaffe都进行了安装，实际操作的童鞋可以根据自己的需要进行选择。（要安装matcaffe必须首先安装matlab，安装步骤不在赘述，要安装pycaffe需要首先安装python，可以参考该链接http://blog.csdn.net/qq_14845119/article/details/52354394）相应的修改如下所示。其中，右边为原始的CommonSettings.props.example，左边为需要修改并保存的的CommonSettings.props。

cpu模式的配置：

gpu模式的配置：

注意，CudaVersion修改为自己电脑的cuda版本（本人电脑的cuda为9.2版本）。

同时注意，由于GPU库除了使用cublas等cuda库之外，还用到了cudnn，默认安装驱动是不安装这个的，因此，需要去官网https://developer.nvidia.com/cudnn下载（需要注册）。下载完成后，进行解压，将相应的bin，include，lib分别放于自己的cuda下面的相应目录中，例如，本人电脑的cuda目录