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 Copy for LibTorchDevice

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.

impl DeviceOps for LibTorchDevice

fn id(&self) -> DeviceId

Returns the device id.

fn inner(&self) -> &Self

Returns the inner device without autodiff enabled.

impl Eq for LibTorchDevice

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

Equality operator ==.

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

Inequality operator !=.

impl StructuralPartialEq for LibTorchDevice