Windows 10(64位)配置Caffe运行环境的基本流程

    进行深度学习可以使用Caffe,我使用windows配置了一下Caffe运行环境,非常简单。这里把官网的配置过程搬运过来,以备参考。

官网地址:https://github.com/BVLC/caffe/tree/windows

  配置环境: Windows 10(64位)+ Visual Studio 2013.

 

1 获取安装文件

  首先在官网地址https://github.com/BVLC/caffe/tree/windows下载压缩包,并解压到想要安装的路径下。

 

2 生成新的配置文件

  将.\windows\CommonSettings.props.example复制一份,并且重命名为.\windows\CommonSettings.props(文件名为CommonSettings.props,前边的.\windows\是路径)。

默认情况下,编译需要CUDA和cuDNN库,这些都可以在配置文件.\windows\CommonSettings.props中进行配置。默认情况下不支持Python,但是也可以在.\windows\CommonSettings.props中更改。Caffe所需要的第三方的依赖项可以通过NuGet自动获取。

 

3 CUDA配置

  从网址https://developer.nvidia.com/cuda-toolkit下载最新的CUDA,本文撰写时已经更新到CUDA 8。如果你不想使用CUDA,或者由于某些原因装不上CUDA,也可以采用CPU_ONLY选项来进行生成,此时你需要在.\windows\CommonSettings.props中设置CpuOnlyBuild选项为true,设置UseCuDNN选项为false。

 

4 cuDNN配置(如果第3步中选择CPU_ONLY,则本步骤可以忽略)

  从网址https://developer.nvidia.com/cudnn下载cuDNN v3或者 cuDNN v4。将下载的压缩文件解压到 %CUDA_PATH%(CUDA安装时生成的环境变量)。也可以解压到其他的路径,并将配置文件的CuDnnPath指向解压的路径。当然,如果仅使用CUDA但不使用cuDNN,可以将配置文件中的UseCuDNN设置为false。

 

5 Python接口配置

  在配置文件中将PythonSupport设置为true。然后下载Miniconda 2.7 64-bit Windows安装文件,为所有用户添加Python路径到PATH环境变量(通过安装文件完成),下载地址为http://conda.pydata.org/miniconda.html

  以“管理员身份”打开命令提示符窗口,输入以下命令:

  conda install --yes numpy scipy matplotlib scikit-image pip

  pip install protobuf

  注意:编译完成后,为了能够使用Python,需要做以下更改:

设置PythonPath环境变量指向<caffe_root>\Build\x64\Release\pycaffe,或者将文件夹<caffe_root>\Build\x64\Release\pycaffe\caffe拷贝到<python_root>\lib\site-packages路径下。

 

6 Matlab接口配置

  在配置文件中将MatlabSupport设置为true,并且将MatlabDir设置为你的Matlab安装的根目录。

  注意:编译完成后为了使用Matlab,需要做以下更改:

  将生成的matcaffe文件夹添加到Matlab的搜索路径,并且,添加<caffe_root>\Build\x64\Release到系统路径。

 

7 编译生成

  打开.\windows\Caffe.sln,全部生成即可。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
要在Windows配置Caffe,首先需要确保已经安装了所需的软件和工具。以下是详细的步骤: 1. 安装CUDA和CUDNN:Caffe在GPU上进行加速需要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,可以使用它进行深度学习任务了。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值