Processing unit settings

task_type

Command line: --task-type

Description

The processing unit type to use for training.

Possible values:

  • CPU
  • GPU

Type

string

Default value

CPU

Supported processing units

CPU and GPU

devices

Command line: --devices

Description

IDs of the GPU devices to use for training (indices are zero-based).

Format

  • <unit ID> for one device (for example, 3)
  • <unit ID1>:<unit ID2>:..:<unit IDN> for multiple devices (for example, devices='0:1:3')
  • <unit ID1>-<unit IDN> for a range of devices (for example, devices='0-3')

Type

string

Default value

Python package

NULL (all GPU devices are used if the corresponding processing unit type is
selected)

R package

-1 (all GPU devices are used if the corresponding processing unit type is selected)

Command-line

-1 (use all devices)

Supported processing units

CPU

In this article: