GPU enabled Docker image for Calamari usage with OCR4all
- Docker
- NVIDIA Container Toolkit
- Local CUDA / CUDNN setup and an nvidia GPU
Build image
docker build . -t calamari:gpu
Run a command
docker run --gpus all --rm calamari:gpu calamari-predict ...
or if you're not on a amd64 platform
docker run --platform linux/amd64 --rm calamari:gpu calamari-predict ...