pub struct GpuConfig {
pub backend: Option<GpuBackend>,
pub device_id: Option<usize>,
pub mixed_precision: bool,
pub tensor_cores: bool,
pub memory_growth: bool,
pub memory_fraction: f32,
}Expand description
GPU configuration
Fields§
§backend: Option<GpuBackend>Preferred backend (None for auto-select)
device_id: Option<usize>Device ID (None for auto-select)
mixed_precision: boolEnable mixed precision (FP16)
tensor_cores: boolEnable tensor cores (CUDA)
memory_growth: boolMemory growth (allocate on demand)
memory_fraction: f32Per-process GPU memory fraction
Implementations§
Trait Implementations§
Auto Trait Implementations§
impl Freeze for GpuConfig
impl RefUnwindSafe for GpuConfig
impl Send for GpuConfig
impl Sync for GpuConfig
impl Unpin for GpuConfig
impl UnsafeUnpin for GpuConfig
impl UnwindSafe for GpuConfig
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more