只需要在python脚本中添加两行代码:
caffe.set_device(0) #使用第一块显卡
caffe.set_mode_gpu() #设为gpu模式
这样,就可以在默认显卡被占用(第一块显卡)时,选择其他显卡进行测试了。
只需要在python脚本中添加两行代码:
caffe.set_device(0) #使用第一块显卡
caffe.set_mode_gpu() #设为gpu模式
这样,就可以在默认显卡被占用(第一块显卡)时,选择其他显卡进行测试了。
转载于:https://www.cnblogs.com/Allen-rg/p/9032513.html