--gpu_id

Function Usage

This option applies only to Caffe networks.

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

See Also

None

Arguments

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

Remarks: Before using this option, ensure that CUDA is installed by following instructions provided in Preparing for Installation.

Default: none.

Suggestions and Benefits

None

Examples

--gpu_id=1

Dependencies and Restrictions

None