Burn Playground

A Rust project for experimenting with the burn deep learning framework.

Note: This code is adapted from burn's example repository. I've only tested it with CUDA.

Setup

# Clone the repository
git clone <repository-url>
cd burn-playground

# Build and run
cargo run --release --features cuda

Features

• MNIST training of a simple convolutional neural network (from burn's repo)
• Docker image

Usage

Run the application using the provided Docker image with CUDA support:

# Build the Docker image
docker build -t burn-playground .

# Run with NVIDIA GPU support
docker run --runtime nvidia -it burn-playground