SetCurrentGpuDeviceId property

Pointer<NativeFunction<OrtStatusPtr Function(Int device_id)>> SetCurrentGpuDeviceId

\brief Set current GPU device ID

Set the current device id of the GPU execution provider (CUDA/tensorrt/rocm). The device id should be less than the total number of devices available. This is only useful when multiple-GPUs are installed and it is required to restrict execution to a single GPU.

\paramin device_id

\snippet{doc} snippets.dox OrtStatus Return Value

Implementation

external ffi.Pointer<
  ffi.NativeFunction<OrtStatusPtr Function(ffi.Int device_id)>
>
SetCurrentGpuDeviceId;