README.md

dfdx_cifar

Classifying the CIFAR-10 dataset using the dfdx library.

# Build the Docker container for dfdx with CUDA support
$ docker build . -t dfdx:latest
# Mount the working directory into the dfdx container at /home
$ docker run --gpus all --rm -it -v $(pwd):/home dfdx:latest
$ cargo run --release --features=dfdx/cuda

Tested on Pop!_OS with an NVIDIA 4070Ti