--gpu_id

Description

This option applies only to Caffe models.

Specifies a GPU for quantization. If this option is not included, quantization is performed on the CPU.

See Also

None

Argument

Argument: ID of an available GPU. Run the nvidia-smi command to query available GPUs.

Restrictions: Before using this option, ensure that CUDA is installed by following instructions provided in Installation Preparations.

Default: none.

Suggestions and Benefits

None

Example

--gpu_id=1

Restrictions

None