Windows下Caffe的安装与配置(GPU版)

零、写在前面

由于Caffe的依赖项相当多,在配置时可能会遇到各种问题,整个过程繁琐而痛苦,需要足够的耐心,请全程保持微笑:)
由于毕业设计,近期正在进行机器学习、深度学习的学习,学到caffe,正好把整个过程进行记录,建议大家在大三时起就主动进行学习,不要像我一样大四才开始知道着急。
好了说到这里,我的电脑配置如下所示。
操作系统:win10家庭中文版
处理器: i5-6300HQ
内存:8G
显卡:GTX 950M 4G独显

一、准备

  1. 首先确认你的电脑是否装有NviDIA的独立显卡,如果没有,查看CPU版的配置过程。

  2. Visual Studio 2013:Caffe仅支持VS2013和VS2015,但在VS2015下问题比较多,所以选择VS2013进行编译和调试。如果安装了更高版本的Visual Studio,先卸载掉。

    *不同版本的VS应该是可以共存的,需要先安装低版本再安装高版本。但是保险起见,先只保留VS2013吧。

  3. caffe-windows:下载地址,直接下载zip解压即可。

  4. NviDIA CUDA和CudNN:
    CUDA:https://developer.nvidia.com/cuda-downloads
    CudNN:https://developer.nvidia.com/cudnn
    首先进入N

  • 2
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
要在Windows配置Caffe,首先需要确保已经安装了所需的软件和工具。以下是详细的步骤: 1. 安装CUDA和CUDNN:CaffeGPU上进行加速需要CUDA和CUDNN的支持。首先,下载并安装适合你的显卡型号的CUDA驱动程序。然后,从NVIDIA的官方网站下载并安装CUDNN库。 2. 安装依赖库:在Windows上运行Caffe还需要一些依赖库。你可以使用包管理工具(如Chocolatey)快速安装这些库。运行命令`choco install git python2`来安装Git和Python。 3. 克隆Caffe:打开命令提示符,导航到你想要存储Caffe的目录,并使用以下命令克隆Caffe的代码库:`git clone https://github.com/BVLC/caffe.git` 4. 配置环境:进入Caffe目录,在根目录下创建一个新的文件夹"build",用于存储编译后的文件。然后,将模板文件夹"windows"复制到"build"文件夹内。重命名此文件夹为"CMake"。 5. 使用CMake配置:打开CMake GUI,选择"Caffe"文件夹作为源代码目录,选择"build"文件夹作为目标目录。点击"Configure"按钮选择你安装的编译器和生成器(如Visual Studio)。然后点击"Generate"生成解决方案。 6. 编译Caffe:打开生成的解决方案,选择合适的编译选项(如Release、Debug等),然后开始编译。这可能需要一些时间,取决于你的电脑配置和编译选项。 7. 配置环境变量:将Caffe的"build\bin"路径添加到系统的环境变量中,这样才能在命令提示符或其他地方运行Caffe命令。 现在,你已经成功在Windows配置Caffe,可以使用它进行深度学习任务了。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值