Adding `burn_cuda` library as a dependency

Run this command in a terminal, in your project's directory:

cargo add burn-cuda

To add it manually, edit your project's Cargo.toml file and add to the [dependencies] section:

burn-cuda = "0.21.0"

The burn_cuda library will be automatically available globally. Read the burn_cuda library documentation.

Back to the crate overview.

Readme

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.

Adding burn_cuda library as a dependency

Adding `burn_cuda` library as a dependency