dokai

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

Example

Pull an image

docker pull ghcr.io/osai-ai/dokai:21.07-pytorch

Docker Hub mirror

docker pull osaiai/dokai:21.07-pytorch

Check available GPUs inside container

docker run --rm \
    --gpus=all \
    ghcr.io/osai-ai/dokai:21.07-pytorch \
    nvidia-smi

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

GitHub

https://github.com/osai-ai/dokai