---
title: --gpu_id
description: "该参数仅适用于Caffe框架网络模型。"
url: https://www.hiascend.com/document/detail/zh/canncommercial/latest/devaids/amct/atlasamct_16_0353.html
sourcePath: /source/zh/canncommercial/900/devaids/amct/atlasamct_16_0353.html
indexId: 9136197ed4f04bd8a675ecb4d50bc2b3105b97aa2495e87b5864b99e308007c165
---
# --gpu_id

#### 功能说明

该参数仅适用于Caffe框架网络模型。

指定使用GPU进行量化，不使用该参数，则默认使用CPU进行量化。


#### 关联参数

无。


#### 参数取值

参数值：可用的GPU ID，用户可以通过nvidia-smi命令查询可用的GPU。

参数值约束：使用该参数时，请确保根据安装前准备已经安装CUDA软件包。

参数默认值：无


#### 推荐配置及收益

无。


#### 示例

```
--gpu_id=1
```


#### 依赖约束

无。