Ben_Kosytorz
3a67c0979c
- Updated stablediffusion crate path from "../stable-diffusion-burn" to "./crates/stable-diffusion-burn" for proper workspace resolution - Enhanced .gitignore to include generated model files (.mpk, .pt, .bin, .safetensors, .ckpt) and user_data directory - Added Cargo.lock to gitignore with appropriate comment - Reorganized IDE files section in gitignore for better clarity - Added newline at end of file for proper formatting
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.