2 unstable releases
0.15.0 | Oct 28, 2024 |
---|---|
0.14.0 | Aug 27, 2024 |
#1066 in Machine learning
1,797 downloads per month
Used in 15 crates
(via burn-core)
2MB
45K
SLoC
Burn CUDA Backend
Burn CUDA backend
This crate provides a CUDA backend for Burn using the cubecl and cudarc crates.
Usage Example
#[cfg(feature = "cuda")]
mod cuda {
use burn_autodiff::Autodiff;
use burn_cuda::{Cuda, CudaDevice};
use mnist::training;
pub fn run() {
let device = CudaDevice::default();
training::run::<Autodiff<Cuda<f32, i32>>>(device);
}
}
Dependencies
Requires CUDA 12.x to be installed and on the PATH
.
Dependencies
~38–73MB
~1.5M SLoC