device_id
产品支持情况
产品 |
是否支持 |
---|---|
√ |
|
√ |
|
x |
|
x |
|
x |
|
x |
函数功能
设置当前进程Device ID,对应底层ge.exec.deviceId配置项。
函数原型
1 | device_id(device_id) |
参数说明
参数名称 |
数据类型 |
取值说明 |
---|---|---|
device_id |
Union[int, List[int], Tuple[int]] |
设置当前进程的Device ID。当前只支持配置一个 |
调用示例
1 2 3 | from llm_datadist import LLMConfig llm_config = LLMConfig() llm_config.device_id = 0 |
返回值
正常情况下无返回值。
参数错误可能抛出TypeError或ValueError。
约束说明
无
父主题: LLMConfig