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

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

　　

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