windows下检验caffe是否配置正确

Posted 2020-09-21 leoking01

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