mirror of https://github.com/tracel-ai/burn.git
30 lines
895 B
Markdown
30 lines
895 B
Markdown
# Burn CUDA Backend
|
|
|
|
[Burn](https://github.com/tracel-ai/burn) CUDA backend
|
|
|
|
[](https://crates.io/crates/burn-cuda)
|
|
[](https://github.com/tracel-ai/burn-cuda/blob/master/README.md)
|
|
|
|
This crate provides a CUDA backend for [Burn](https://github.com/tracel-ai/burn) using the
|
|
[cubecl](https://github.com/tracel-ai/cubecl.git) and [cudarc](https://github.com/coreylowman/cudarc.git)
|
|
crates.
|
|
|
|
## Usage Example
|
|
|
|
```rust
|
|
#[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`. |