---
title: set_gpu_mode
description: "| 产品 | 是否支持 |"
url: https://www.hiascend.com/document/detail/zh/canncommercial/latest/devaids/amct/atlasamct_16_0268.html
sourcePath: /source/zh/canncommercial/900/devaids/amct/atlasamct_16_0268.html
indexId: ccd2a42ecfd1e20b9ba26d12a25d7163f4e215e0aa22ad7d9306d1d42228e86765
---
# set_gpu_mode

#### 产品支持情况

| 产品 | 是否支持 |
| --- | --- |
| Atlas 350 加速卡 | x |
| Atlas A3 训练系列产品 / Atlas A3 推理系列产品 | x |
| Atlas A2 训练系列产品 / Atlas A2 推理系列产品 | x |
| Atlas 200I/500 A2 推理产品 | √ |
| Atlas 推理系列产品 | √ |
| Atlas 训练系列产品 | √ |


#### 功能说明

调用该接口之后，AMCT执行权重量化的时候，会使用GPU进行加速。


#### 函数原型

```
set_gpu_mode()
```


#### 参数说明

无


#### 返回值说明

无


#### 约束说明

- 用户有GPU环境，且支持CUDA10.0。该接口不支持选择GPU卡，用户可以通过CUDA环境变量(CUDA_VISIBLE_DEVICES)来选择GPU卡，或者使用pycaffe的set_device()接口来选择GPU卡。
- 如果既不调用  set_gpu_mode
接口也不调用  set_cpu_mode
接口。AMCT默认使用CPU进行权重的量化。


#### 调用示例

```
import amct_caffe as amct
amct.set_gpu_mode()
```