burn/crates/burn-cuda
Nathaniel Simard d9e41460ff
Refactor burn jit => burn-cubecl (#2809)
2025-02-13 12:39:29 -05:00
..
src Refactor burn jit => burn-cubecl (#2809) 2025-02-13 12:39:29 -05:00
Cargo.toml Rename crate 2025-02-13 11:59:09 -05:00
README.md Add documentation for custom `cubecl` kernels, update some outdated docs (#2404) 2024-10-25 13:22:23 -04:00

README.md

Burn CUDA Backend

Burn CUDA backend

Current Crates.io Version license

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.