windows下检验caffe是否配置正确

windows下检验caffe是否配置正确:(注:不考虑搭建caffe的编译环境,而是直接使用caffe官网提供的二进制文件)

windows版本源码以及二进制库文件下载地址:https://github.com/BVLC/caffe/tree/windows

#encoding: utf-8

import numpy as np;

import sys,os;

# 注意下面的目录结构,根据自己的实际情况修改即可

caffe_root='D:/env/Caffe/caffe/';

sys.path.insert(0, caffe_root + 'python');

import caffe as ca; # 引入自己的caffe的python接口

print("######"); # python打印6个#

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();

print ("finish set_mode_gpu.")

# 下面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;

print ("all ok")

依赖安装说明:

使用anaconda4.2.0,即python3.5,实际上已经继承了numpy,scipy,sklearn等一干工具。

还缺少的依赖有:pydot, 但是在python3.5平台下建议使用pydotplus, 而不是pydot,否则会报错。

另外在安装pydotplus的基础上使用pip重新安装graphviz。

备注:对这里的检验来说,只要用pip安装graphviz即可。但是在有些情况(绝大多数情况下)下,应当独立安装graphviz工具才可以。下载地址:http://www.graphviz.org/Download_windows.php