二次开发如何指定显卡

[aqrose]

add_model_engine(“模块类型”, “模型地址”, “模型定制参数”)

该函数中模型定制参数为test.json的定制版本，可以通过json解析库（jsoncpp）解析test.json，然后修改gpu_ids参数为指定的GPU编号，然后保存为test-{gpu编号}.json，然后将test-{gpu编号}.json传递给“模型定制参数”

[nana.luo]

在test.json内修改gpu_ids中的”value”参数指定GPU，如下：

“gpu_ids”: {
“annotate”: “优先使用指定的GPU, 如果指定GPU少于GPU数量，将根据情况自动分配其他GPU”,
“cn_name”: “指定GPU”,
“data_type”: “ph_VecInt”,
“id”: 5,
“meta_type”: 19,
“permission”: 1,
“value”: null
},

gpu_ids的value是一个数组，应设为[0]、[1]等；如设为0、1则指定GPU不起作用。

查看日志里的警告可有助于定位该类问题。

[作者]

帖子