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

只需要在python脚本中添加两行代码:
caffe.set_device(0)   #使用第一块显卡
caffe.set_mode_gpu()  #设为gpu模式

  

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