CUDA

data class CUDA(deviceId: Int) : ExecutionProvider
Content copied to clipboard

CUDA execution provider. Available only on platforms with Nvidia gpu and CUDA runtime installed.

See also

OrtProvider.CUDA

Parameters

deviceId

The device ID to use.

Constructors

CUDA
Link copied to clipboard
common
fun CUDA(deviceId: Int = 0)
Content copied to clipboard

Functions

addOptionsTo
Link copied to clipboard
common
open override fun addOptionsTo(sessionOptions: <ERROR CLASS>)
Content copied to clipboard

Adds execution provider options to the OrtSession.SessionOptions.

Properties

deviceId
Link copied to clipboard
common
val deviceId: Int = 0
Content copied to clipboard
internalProviderId
Link copied to clipboard
common
val internalProviderId: <ERROR CLASS>
Content copied to clipboard