Skip to content

Releases: osai-ai/dokai

v24.06

18 Jul 07:21
472865c
Compare
Choose a tag to compare

What's Changed

  • Add Dokai images mirroring;
  • Add .rootless images;
  • Fix ffmpeg and .opt.rootless images;
  • Add cpu.pytorch.* images;
  • Add more automatic tests;
  • Introduce performance, time, and size inspection scripts;
  • Update CI/CD version;
  • Updated README.md;

Full Changelog: v24.04...v24.06

v24.04: Add CPU-based images

23 Apr 08:44
01fec3b
Compare
Choose a tag to compare

Updates

  • Added CPU-based images;
  • Refactored GPU-based images, changed their names and moved out apt-packages;

v24.03: PyTorch (2.1.2), TensorRT (9.2.0), FFmpeg (6.1)

04 Apr 12:52
652f53f
Compare
Choose a tag to compare

Updates

  • Added .opt docker images set which is optimized in size;
  • Split base image to ffmpeg and base;
  • Removed librosa, numba, llvm, av, cupy-cuda12x, pytorch-ignite;

PyTorch (2.1.2), TensorRT (9.2.0), FFmpeg (6.1)

20 Dec 17:32
Compare
Choose a tag to compare

Updates

  • PyTorch (2.1.2), torchvision (0.16.2), torchaudio (2.1.2)
  • TensorRT (9.2.0)
  • FFmpeg (6.1)
  • Python (3.11.7)
  • cuDNN (8.9.6)
  • Update other PyPI packages
  • Remove Torch-TensorRT

Images

CUDA (12.2.2), cuDNN (8.9.6), Python (3.11.7)
ghcr.io/osai-ai/dokai:23.12-core

dokai:23.12-core

CUDA (12.2.2), cuDNN (8.9.6)
CMake (3.22.1)
Python (3.11.7)

pip==23.3.2
setuptools==69.0.2

Python ML and CV packages, FFmpeg (6.1) with NVENC/NVDEC support
ghcr.io/osai-ai/dokai:23.12-base

dokai:23.12-base

additionally to dokai:23.12-core:

FFmpeg (n6.1), nv-codec-headers (n12.1.14.0)

packaging==23.2
numpy==1.26.2
opencv-python==4.8.1.78
sympy==1.12
scipy==1.11.4
matplotlib==3.8.2
pandas==2.1.4
scikit-learn==1.3.2
scikit-image==0.22.0
Pillow==10.1.0
av==11.0.0
librosa==0.10.1
albumentations==1.3.1
pyzmq==25.1.2
Cython==3.0.6
numba==0.58.1
requests==2.31.0
psutil==5.9.7
pydantic==2.5.2
PyYAML==6.0.1
notebook==7.0.6
ipywidgets==8.1.1
tqdm==4.66.1
pytest==7.4.3
pytest-cov==4.1.0
mypy==1.7.1
flake8==6.1.0
pre-commit==3.6.0

TensorRT (9.2.0), PyTorch (2.1.2), torchvision (0.16.2), torchaudio (2.1.2) and torch based libraries
ghcr.io/osai-ai/dokai:23.12-pytorch

dokai:23.12-pytorch

additionally to dokai:23.12-base:

TensorRT (9.2.0)
MAGMA (2.7.1)

torch==2.1.2 (source, v2.1.2)
torchvision==0.16.2 (source, v0.16.2)
torchaudio==2.1.2 (source, v2.1.2)
triton==2.1.0
cupy-cuda12x==12.3.0
pytorch-ignite==0.4.13
pytorch-argus==1.0.0
kornia==0.7.0
timm==0.9.12
onnx==1.15.0
onnxruntime==1.16.3
onnxsim==0.4.35

VideoProcessingFramework (2.0.0) and Tensor Stream (0.5.0) for full HW acceleration of video decoding
ghcr.io/osai-ai/dokai:23.12-video

dokai:23.12-video

additionally to dokai:23.12-pytorch:

VideoProcessingFramework==2.0.0 (source, v2.0.0)
tensor-stream==0.5.0 (source, 0.5.0)

Hopper architecture support, CUDA (12.2.2), Python (3.11.6), PyTorch (2.1.0)

27 Oct 23:31
Compare
Choose a tag to compare

Updates

  • CUDA (12.2.2), cuDNN (8.9.5)
  • Python (3.11.6)
  • PyTorch (2.1.0)
  • PyAV (10.0.0)
  • Update other PyPI packages

Images

CUDA (12.2.2), cuDNN (8.9.5), Python (3.11.6)
ghcr.io/osai-ai/dokai:23.10-core

dokai:23.10-core

CUDA (12.2.2), cuDNN (8.9.5)
Python (3.11.6)
CMake (3.22.1)
Bazelisk (1.18.0)

Python ML and CV packages, FFmpeg (6.0) with NVENC/NVDEC support
ghcr.io/osai-ai/dokai:23.10-base

dokai:23.10-base

additionally to dokai:23.10-core:

FFmpeg (n6.0), nv-codec-headers (n12.1.14.0)

packaging==23.2
numpy==1.26.1
opencv-python==4.8.1.78
sympy==1.12
scipy==1.11.3
matplotlib==3.8.0
pandas==2.1.1
scikit-learn==1.3.2
scikit-image==0.22.0
Pillow==10.1.0
av==10.0.0
librosa==0.10.1
albumentations==1.3.1
pyzmq==25.1.1
Cython==3.0.4
numba==0.58.1
requests==2.31.0
psutil==5.9.6
pydantic==2.4.2
PyYAML==6.0.1
notebook==7.0.6
ipywidgets==8.1.1
tqdm==4.66.1
pytest==7.4.2
pytest-cov==4.1.0
mypy==1.6.1
flake8==6.1.0
pre-commit==3.5.0

TensorRT (8.6.1), PyTorch (2.1.0), torchvision (0.16.0), torchaudio (2.1.0) and torch based libraries
ghcr.io/osai-ai/dokai:23.10-pytorch

dokai:23.10-pytorch

additionally to dokai:23.10-base:

TensorRT (8.6.1)
MAGMA (2.7.1)

torch==2.1.0 (source, v2.1.0)
torch_tensorrt (source, v2.1.0-rc3)
torchvision==0.16.0 (source, v0.16.0)
torchaudio==2.1.0 (source, v2.1.0)
triton==2.1.0
cupy-cuda12x==12.2.0
pytorch-ignite==0.4.13
pytorch-argus==1.0.0
kornia==0.7.0
timm==0.9.8
onnx==1.14.1
onnxruntime==1.16.1
onnxsim==0.4.35

VideoProcessingFramework (2.0.0) and Tensor Stream (0.5.0) for full HW acceleration of video decoding
ghcr.io/osai-ai/dokai:23.10-video

dokai:23.10-video

additionally to dokai:23.10-pytorch:

VideoProcessingFramework==2.0.0 (source, v2.0.0)
tensor-stream==0.5.0 (source, 0.5.0)

VideoProcessingFramework (2.0.0), PyTorch (2.0.1), timm (0.9.2), FFmpeg (6.0), TensorRT (8.6.1)

23 May 08:20
Compare
Choose a tag to compare

Updates

  • PyTorch (2.0.1)
  • TensorRT (8.6.1)
  • Torch-TensorRT (v1.4.0)
  • timm (0.9.2)
  • VideoProcessingFramework (2.0.0)
  • TensorStream (0.5.0)
  • FFmpeg (6.0)
  • Update other PyPI packages

Images

base

Python with ML and CV packages, CUDA (11.8.0), cuDNN (8.7.0), FFmpeg (6.0) with NVENC/NVDEC support
ghcr.io/osai-ai/dokai:23.05-base

dokai:23.05-base

Supported NVIDIA architectures: Pascal (sm_60, sm_61), Volta (sm_70), Turing (sm_75), Ampere (sm_80, sm_86), Ada Lovelace (sm_89).

CUDA (11.8.0), cuDNN (8.7.0)
FFmpeg (n6.0), nv-codec-headers (n12.0.16.0)
Python (3.10.6)
CMake (3.22.1)

pip==23.1.2
setuptools==67.7.2
packaging==23.1
numpy==1.24.3
opencv-python==4.7.0.72
sympy==1.12
scipy==1.10.1
matplotlib==3.7.1
pandas==2.0.1
scikit-learn==1.2.2
scikit-image==0.20.0
Pillow==9.5.0
librosa==0.10.0.post2
albumentations==1.3.0
pyzmq==25.0.2
Cython==0.29.34
numba==0.57.0
requests==2.30.0
psutil==5.9.5
pydantic==1.10.7
PyYAML==6.0
notebook==6.5.2
ipywidgets==8.0.6
tqdm==4.65.0
pytest==7.3.1
pytest-cov==4.0.0
mypy==1.3.0
flake8==6.0.0
pre-commit==3.3.1

pytorch

TensorRT (8.6.1), PyTorch (2.0.1), torchvision (0.15.2), torchaudio (2.0.2) and torch based libraries.
ghcr.io/osai-ai/dokai:23.05-pytorch

dokai:23.05-pytorch

additionally to dokai:23.05-base:

TensorRT (8.6.1)
MAGMA (2.7.1)

torch==2.0.1 (source, v2.0.1)
torch_tensorrt (source, release/1.4)
torchvision==0.15.2 (source, v0.15.2 tag)
torchaudio==2.0.2 (source, v2.0.2 tag)
pytorch-ignite==0.4.12
pytorch-argus==1.0.0
pretrainedmodels==0.7.4
efficientnet-pytorch==0.7.1
pytorch-toolbelt==0.6.3
kornia==0.6.12
timm==0.9.2

tensor-stream

Tensor Stream (0.5.0) for real-time video streams decoding on GPU.
ghcr.io/osai-ai/dokai:23.05-tensor-stream

dokai:23.05-tensor-stream

additionally to dokai:23.05-pytorch:

tensor-stream==0.5.0 (source, 0.5.0 tag)

vpf

VideoProcessingFramework (2.0.0) for full HW acceleration for video decoding and encoding.
ghcr.io/osai-ai/dokai:23.05-vpf

dokai:23.05-tensor-stream

additionally to dokai:23.05-pytorch:

VideoProcessingFramework (source, v2.0.0 tag)

Ada Lovelace architecture support, CUDA 11.8.0, cuDNN 8.6.0, TensorRT 8.5.1, PyTorch 1.13.0

22 Nov 19:11
Compare
Choose a tag to compare

Updates

  • TensorRT 8.5.1
  • torch 1.14.0a0+git71fe069 (source, close to v1.13.0 after commit "ada lovelace (arch 8.9) support #87436")
  • torchvision 0.14.0 (from source, v0.14.0 tag)
  • torchaudio 0.13.0 (from source, v0.13.0 tag)
  • Update other PyPI packages
  • Ada Lovelace architecture support
  • PyTorch image models benchmark link

Images

base

Python with ML and CV packages, CUDA (11.8.0), cuDNN (8.6.0), FFmpeg (4.4) with NVENC/NVDEC support
ghcr.io/osai-ai/dokai:22.11-base

dokai:22.11-base

Supported NVIDIA architectures: Pascal (sm_60, sm_61), Volta (sm_70), Turing (sm_75), Ampere (sm_80, sm_86), Ada Lovelace (sm_89).

CUDA (11.8.0), cuDNN (8.6.0)
FFmpeg (release/4.4), nv-codec-headers (sdk/11.0)
Python (3.10.6)
CMake (3.22.1)

pip==22.3.1
setuptools==65.5.1
packaging==21.3
numpy==1.23.4
opencv-python==4.6.0.66
scipy==1.9.3
matplotlib==3.6.2
pandas==1.5.1
scikit-learn==1.1.3
scikit-image==0.19.3
Pillow==9.3.0
librosa==0.9.2
albumentations==1.3.0
pyzmq==24.0.1
Cython==0.29.32
numba==0.56.4
requests==2.28.1
psutil==5.9.4
pydantic==1.10.2
PyYAML==6.0
notebook==6.5.2
ipywidgets==8.0.2
tqdm==4.64.1
pytest==7.2.0
pytest-cov==4.0.0
mypy==0.991
flake8==5.0.4
pre-commit==2.20.0

pytorch

TensorRT (8.5.1) , PyTorch (1.13.0), torchvision (0.14.0), torchaudio (0.13.0) and torch based libraries.
ghcr.io/osai-ai/dokai:22.11-pytorch

dokai:22.11-pytorch

additionally to dokai:22.11-base:

TensorRT (8.5.1)
MAGMA (2.6.2)

torch==1.14.0a0+git71fe069 (source, close to v1.13.0 after commit "ada lovelace (arch 8.9) support #87436")
torchvision==0.14.0 (source, v0.14.0 tag)
torchaudio==0.13.0 (source, v0.13.0 tag)
pytorch-ignite==0.4.10
pytorch-argus==1.0.0
pretrainedmodels==0.7.4
efficientnet-pytorch==0.7.1
pytorch-toolbelt==0.5.2
kornia==0.6.8
timm==0.6.11
segmentation-models-pytorch==0.3.0

tensor-stream

Tensor Stream for real-time video streams decoding on GPU.
ghcr.io/osai-ai/dokai:22.11-tensor-stream

dokai:22.11-tensor-stream

additionally to dokai:22.11-pytorch:

tensor-stream==0.4.6 (source, dev branch)

CUDA 11.6.0, PyTorch 1.11.0

28 Mar 14:03
61cf105
Compare
Choose a tag to compare

Updates

  • CUDA 11.6.0
  • torch 1.11.0 (from source, v1.11.0 tag)
  • torchvision 0.12.0 (from source, v0.12.0 tag)
  • torchaudio 0.11.0 (from source, v0.11.0 tag)
  • CMake (3.22.2)
  • Update other PyPI packages
  • Update README

Images

base

Python with ML and CV packages, CUDA (11.6.0), FFmpeg (4.4) with NVENC support.

dokai:22.03-base

ghcr.io/osai-ai/dokai:22.03-base

CUDA (11.6.0)
FFmpeg (release/4.4), nv-codec-headers (sdk/11.0)
Python (3.8.10)
CMake (3.22.2)

pip==22.0.3
setuptools==59.5.0
packaging==21.3
numpy==1.21.5
opencv-python==4.5.5.62
scipy==1.8.0
matplotlib==3.5.1
pandas==1.4.1
scikit-learn==1.0.1
scikit-image==0.18.3
Pillow==8.4.0
librosa==0.8.1
albumentations==1.1.0
pyzmq==22.3.0
Cython==0.29.24
numba==0.53.1
requests==2.26.0
psutil==5.8.0
pydantic==1.8.2
PyYAML==6.0
notebook==6.4.5
ipywidgets==7.6.5
tqdm==4.62.3
pytest==6.2.5
mypy==0.910
flake8==4.0.1

pytorch

PyTorch, torchvision and torch based libraries.

dokai:22.03-pytorch

ghcr.io/osai-ai/dokai:22.03-pytorch

additionally to dokai:22.03-base:

MAGMA (2.6.1)

torch==1.11.0 (source, v1.11.0 tag)
torchvision==0.12.0 (source, v0.12.0 tag)
torchaudio==0.11.0 (source, v0.11.0 tag)
pytorch-ignite==0.4.8
pytorch-argus==1.0.0
pretrainedmodels==0.7.4
efficientnet-pytorch==0.7.1
timm==0.5.4
segmentation-models-pytorch==0.2.1
kornia==0.6.3

tensor-stream

Tensor Stream.

dokai:22.03-tensor-stream

ghcr.io/osai-ai/dokai:22.03-tensor-stream

additionally to dokai:22.03-pytorch:

tensor-stream==0.4.6 (source, dev branch)

PyTorch 1.10.0, CMake 3.21.4

09 Nov 21:40
Compare
Choose a tag to compare

Updates

  • torch 1.10.0 (from source, v1.10.0 tag)
  • torchvision 0.11.1 (from source, v0.11.1 tag)
  • torchaudio 0.10.0 (from source, v0.10.0 tag)
  • CMake (3.21.4)
  • Remove Apex installation
  • Update other PyPI packages

Images

base

Python with ML and CV packages, CUDA (11.4.2), cuDNN (8.2.4), FFmpeg (4.4) with NVENC support.
ghcr.io/osai-ai/dokai:21.11-base

dokai:21.11-base

CUDA (11.4.2), cuDNN (8.2.4)
FFmpeg (release/4.4), nv-codec-headers (sdk/11.0)
Python (3.8.10)
CMake (3.21.4)

pip==21.3.1
setuptools==58.5.3
packaging==21.2
numpy==1.21.4
opencv-python==4.5.4.58
scipy==1.7.2
matplotlib==3.4.3
pandas==1.3.4
scikit-learn==1.0.1
scikit-image==0.18.3
Pillow==8.4.0
librosa==0.8.1
albumentations==1.1.0
pyzmq==22.3.0
Cython==0.29.24
numba==0.53.1
requests==2.26.0
psutil==5.8.0
pydantic==1.8.2
PyYAML==6.0
notebook==6.4.5
ipywidgets==7.6.5
tqdm==4.62.3
pytest==6.2.5
mypy==0.910
flake8==4.0.1

pytorch

PyTorch, torchvision, Apex and torch based libraries.
ghcr.io/osai-ai/dokai:21.11-pytorch

dokai:21.11-pytorch

additionally to dokai:21.11-base:

MAGMA (2.6.1)

torch==1.10.0 (source, v1.10.0 tag)
torchvision==0.11.1 (source, v0.11.1 tag)
torchaudio==0.10.0 (source, v0.10.0 tag)
pytorch-ignite==0.4.7
pytorch-argus==1.0.0
pretrainedmodels==0.7.4
efficientnet-pytorch==0.7.1
timm==0.4.12
segmentation-models-pytorch==0.2.0
kornia==0.6.1

tensor-stream

Tensor Stream.
ghcr.io/osai-ai/dokai:21.11-tensor-stream

dokai:21.11-tensor-stream

additionally to dokai:21.11-pytorch:

tensor-stream==0.4.6 (source, dev branch)

CUDA 11.4.2, cuDNN 8.2.4, PyTorch 1.10.0-rc1

30 Sep 20:47
Compare
Choose a tag to compare

Updates

  • CUDA 11.4.2, cuDNN 8.2.4
  • Build torch 1.10.0-rc1 (from source, v1.10.0-rc1 tag)
  • FFmpeg with HTTPS support
  • kornia 0.5.11
  • Update other PyPI packages

Images

base

Python with ML and CV packages, CUDA (11.4.2), cuDNN (8.2.4), FFmpeg (4.4) with NVENC support.
ghcr.io/osai-ai/dokai:21.09-base

dokai:21.09-base

CUDA (11.4.2), cuDNN (8.2.4)
FFmpeg (release/4.4), nv-codec-headers (sdk/11.0)
Python (3.8.10)

pip==21.2.4
setuptools==58.1.0
packaging==21.0
numpy==1.21.2
opencv-python==4.5.3.56
scipy==1.7.1
matplotlib==3.4.3
pandas==1.3.3
scikit-learn==1.0
scikit-image==0.18.3
Pillow==8.3.2
librosa==0.8.1
albumentations==1.0.3
pyzmq==22.3.0
Cython==0.29.24
numba==0.53.1
requests==2.26.0
psutil==5.8.0
pydantic==1.8.2
PyYAML==5.4.1
notebook==6.4.4
ipywidgets==7.6.5
tqdm==4.62.3
pytest==6.2.5
mypy==0.910
flake8==3.9.2

pytorch

PyTorch, torchvision, Apex and torch based libraries.
ghcr.io/osai-ai/dokai:21.09-pytorch

dokai:21.09-pytorch

additionally to dokai:21.09-base:

MAGMA (2.6.1)

torch==1.10.0-rc1 (source, v1.10.0-rc1 tag)
torchvision==0.10.1 (source, v0.10.1 tag)
torchaudio==0.9.1 (source, v0.9.1 tag)
pytorch-ignite==0.4.6
pytorch-argus==0.2.1
pretrainedmodels==0.7.4
efficientnet-pytorch==0.7.1
timm==0.4.12
segmentation-models-pytorch==0.2.0
kornia==0.5.11
apex (source, master branch)

tensor-stream

Tensor Stream.
ghcr.io/osai-ai/dokai:21.09-tensor-stream

dokai:21.09-tensor-stream

additionally to dokai:21.09-pytorch:

tensor-stream==0.4.6 (source, dev branch)