Collection of Docker images for ML/DL and video processing projects.

Overview of images

Three types of images differ by tag postfix:

  • base: Python with ML and CV packages, CUDA (11.3.1), cuDNN (8.2.0), FFmpeg (4.4) with NVENC support
  • pytorch: PyTorch (1.9.0), torchvision (0.10.0), torchaudio (0.9.0) and torch based libraries
  • tensor-stream: Tensor Stream for real-time video streams decoding on GPU


Pull an image

docker pull

Docker Hub mirror

docker pull osaiai/dokai:21.07-pytorch

Check available GPUs inside container

docker run --rm \
    --gpus=all \ \

Example of using dokai image for DL pipeline you can find here.