使用caffe 的 python接口测试数据，选定GPU编号

只需要在python脚本中添加两行代码：

caffe.set_device(0)   #使用第一块显卡
caffe.set_mode_gpu()  #设为gpu模式

　　

这样，就可以在默认显卡被占用（第一块显卡）时，选择其他显卡进行测试了。