kschuerholt / pytorch_cuda_opencv_ffcv_docker

This repo contains a Dockerfile to build a docker image with opencv with cuda enabled, pytorch and ffcv.

Pytorch Cuda Opencv FFCV Docker

This repo contains a Dockerfile to build a docker image with opencv with cuda enabled, pytorch and ffcv.

Why this docker

FFCV provides a powerful drop-in replacement for datasets and dataloaders, to significantly speed up training. However, it requires opencv as dependency. Installing opencv via apt breaks cuda / mpi, which defeats the purpose. Conda may be an option, but is not always available.

What this docker image does

The docker image provided here extends the NVIDIA Pytorch docker with opencv + ffcv. Opencv is build from source to enable cuda support.