设置当前进程device ID,对应底层ge.exec.deviceId配置项。
device_id(device_id)
参数名称
数据类型
取值说明
device_id
int
设置当前进程的device ID。
1 2 3
from llm_datadist import LLMConfig llm_config = LLMConfig() llm_config.device_id = 0
无