已创建论坛回复

正在查看 2 个帖子:1-2 (共 2 个帖子)
  • 作者
    帖子
  • nana.luo
    参与者

    增加aidi.log会报CUBLAS_STATUS_NOT_INITIALIZED,采用线程池管理可解决报错

    回复至: 二次开发如何指定显卡 #232
    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不起作用。

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

正在查看 2 个帖子:1-2 (共 2 个帖子)