Caffe框架的日常

[Hydraz320]

伤心啊 买来电脑死活装不上ubuntu

[Hydraz320]

Win10系统配置Caffe环境

---

当初调研不力+水平有限，买来的主机竟然一直没装上ubuntu，所以之前给导师交差都用的TensorFlow + Python3.5。然而可以预见的是，但凡有名的模型我都没办法拉下来跑，总之恶心的事情太多，无奈之下，我决定在Win10下把caffe配置起来。不然毕业无望啊。

这里整理下配置环境的步骤细节，我好歹也算是个程序员，好不容易弄好了不写下来还是人？

下载Prebuilt binaries

在经历了一万个恶心的尝试后，我决定使用最暴力的解决方法。首先到官方仓库，我是下载的Visual Studio 2015,CUDA 8.0, Python 2.7这个版本。下载下来后就解压缩就好，里面有目录：bin、include、lib、python、share。其中python是python接口目录。

安装CUDA8.0、cuDNN什么的

这里略去不表，讲的文章很多的。

安装Anaconda-Python2.7

包括一些常用的包什么numpy啦matplotlib啦什么的都安装上。

修改Anaconda的镜像源

看这里吧。https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/

Python接口调用

import numpy as np;
import sys,os;
# 注意下面的目录结构，根据自己的实际情况修改即可
# 总之就是你在我的电脑里输入后能到之前下载下来的caffe的根目录即可 注意最后的斜杠一定要写
caffe_root='C:/hydraz320/paper_work/caffe/caffe-windows/caffe/';

# 来到python接口目录下
sys.path.insert(0, caffe_root + 'python'); 
import caffe as ca; # 引入自己的caffe的python接口
print("######");

import matplotlib.pyplot as plt;
plt.plot([1,2,3,4], [1,2,3,4]);
plt.show();  # 画一条45°直线

ca.set_mode_gpu(); #设置使用GPU，ca.set_mode_cpu();

# 下面5行验证python目录下的接口可以正常引用
import caffe.io as cai;
import caffe.net_spec as can;
import caffe.classifier as cac;
import caffe.draw as cad;  # 这个文件需要安装pydot组件
import caffe.coord_map as caco;

pydot组件安装

首先把环境切换到想要的环境中(注意搜索相关资料先创建一个环境再说)activate caffe，然后先conda install graphviz,再conda install pydot-ng pydotplus即可。这样上面报的错就没有啦。