Enum LibTorchDevice 

pub enum LibTorchDevice {
    Cpu,
    Cuda(usize),
    Mps,
    Vulkan,
}

The device struct when using the tch backend.

Note that you need to provide the device index when using Cuda.

Example

use burn_tch::LibTorchDevice;

let device_gpu_1 = LibTorchDevice::Cuda(0); // First GPU
let device_gpu_2 = LibTorchDevice::Cuda(1); // Second GPU
let device_cpu = LibTorchDevice::Cpu; // CPU
let device_mps = LibTorchDevice::Mps; // Metal Performance Shaders
let device_vulkan = LibTorchDevice::Vulkan; // Vulkan

Variants

Cpu

CPU device.

Cuda(usize)

Cuda device with the given index. The index is the index of the Cuda device in the list of all Cuda devices found on the system.

Mps

Metal Performance Shaders device.

Vulkan

Vulkan device.

Trait Implementations

impl Clone for LibTorchDevice

fn clone(&self) -> LibTorchDevice

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for LibTorchDevice

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for LibTorchDevice

fn default() -> LibTorchDevice

Returns the “default value” for a type.

impl Device for LibTorchDevice

fn from_id(device_id: DeviceId) -> Self

Create a device from its id.

fn to_id(&self) -> DeviceId

Retrieve the device id from the device.

fn device_count(_type_id: u16) -> usize

Returns the number of devices available under the provided type id.

fn device_count_total() -> usize

Returns the total number of devices that can be handled by the runtime.

impl DeviceOps for LibTorchDevice

fn id(&self) -> DeviceId

Returns the device id.

impl From<Device> for LibTorchDevice

fn from(device: Device) -> Self

Converts to this type from the input type.

impl From<LibTorchDevice> for Device

fn from(device: LibTorchDevice) -> Self

Converts to this type from the input type.

impl PartialEq for LibTorchDevice

fn eq(&self, other: &LibTorchDevice) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Copy for LibTorchDevice

impl Eq for LibTorchDevice

impl StructuralPartialEq for LibTorchDevice