Function set_device

pub fn set_device(device: i32) -> Result<()>

Sets a device and initializes it for the current thread.

Parameters

• device: System index of a CUDA device starting with 0.

If the call of this function is omitted, a default device is initialized at the fist CUDA usage.