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 supportpytorch
: PyTorch (1.9.0), torchvision (0.10.0), torchaudio (0.9.0) and torch based librariestensor-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.