首页 > 代码库 > windows下检验caffe是否配置正确
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
windows下检验caffe是否配置正确