diff --git a/python/examples/pipeline/ocr/config.yml b/python/examples/pipeline/ocr/config.yml index 18b960b09adf7647d35b9358fdc5c6dc3aee872a..dca67737472e37fdfbb3fe038159964860b218f9 100644 --- a/python/examples/pipeline/ocr/config.yml +++ b/python/examples/pipeline/ocr/config.yml @@ -40,7 +40,7 @@ op: fetch_list: ["concat_1.tmp_0"] #计算硬件ID,当devices为""或不写时为CPU预测;当devices为"0", "0,1,2"时为GPU预测,表示使用的GPU卡 - devices: "" + devices: "0" #use_mkldnn #use_mkldnn: True @@ -75,7 +75,7 @@ op: device_type: 1 #计算硬件ID,当devices为""或不写时为CPU预测;当devices为"0", "0,1,2"时为GPU预测,表示使用的GPU卡 - devices: "" + devices: "0" #use_mkldnn #use_mkldnn: True