Skip to content

ci(pi): parent image updates #199

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
andhreljaKern merged 8 commits into from
Jun 17, 2026
Merged

ci(pi): parent image updates #199

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
1a4a0d1
ci: update requirements and Dockerfile
Jun 12, 2026
cad8c0e
ci: update requirements.txt
Jun 12, 2026
0eac453
ci: update requirements and Dockerfile
Jun 15, 2026
c05a275
ci(fix): update requirements
andhreljaKern Jun 16, 2026
eec5dee
ci: update requirements and Dockerfile
Jun 16, 2026
01b380f
ci: update requirements.txt
Jun 16, 2026
592c59b
ci: update requirements
andhreljaKern Jun 16, 2026
59566a5
ci(pi): kernai/refinery-parent-images:v3.1.0-torch-cpu
Jun 17, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
ARG PARENT_IMAGE=kernai/refinery-parent-images:v3.0.0-torch-cpu
ARG PARENT_IMAGE=kernai/refinery-parent-images:v3.1.0-torch-cpu
ARG DHI_PYTHON_BUILD=dhi.io/python:3.11-debian12-dev

FROM ${PARENT_IMAGE} AS venv-source
Expand Down
123 changes: 70 additions & 53 deletions gpu-requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@ annotated-types==0.7.0
# via
# -r requirements/torch-cuda-requirements.txt
# pydantic
anyio==4.13.0
anyio==4.14.0
# via
# -r requirements/torch-cuda-requirements.txt
# httpx
Expand Down Expand Up @@ -47,7 +47,7 @@ catalogue==2.0.10
# spacy
# srsly
# thinc
certifi==2026.4.22
certifi==2026.5.20
# via
# -r requirements/torch-cuda-requirements.txt
# httpcore
Expand All @@ -62,9 +62,10 @@ charset-normalizer==3.4.7
# via
# -r requirements/torch-cuda-requirements.txt
# requests
click==8.4.0
click==8.4.1
# via
# -r requirements/torch-cuda-requirements.txt
# huggingface-hub
# typer
# uvicorn
cloudpathlib==0.24.0
Expand All @@ -74,21 +75,27 @@ confection==1.3.3
# spacy
# thinc
# weasel
cuda-bindings==13.3.1
# via torch
cuda-pathfinder==1.5.5
# via cuda-bindings
cuda-toolkit[cudart,cufft,cufile,cupti,curand,cusolver,cusparse,nvjitlink,nvrtc,nvtx]==13.0.2
# via torch
cymem==2.0.13
# via
# preshed
# spacy
# thinc
distro==1.9.0
# via openai
fastapi==0.120.1
fastapi==0.136.3
# via -r requirements/torch-cuda-requirements.txt
filelock==3.20.3
filelock==3.29.4
# via
# -r requirements/torch-cuda-requirements.txt
# huggingface-hub
# torch
fsspec==2026.4.0
fsspec==2026.6.0
# via
# -r requirements/torch-cuda-requirements.txt
# huggingface-hub
Expand All @@ -97,9 +104,9 @@ googleapis-common-protos==1.75.0
# via
# -r requirements/torch-cuda-requirements.txt
# opentelemetry-exporter-otlp-proto-grpc
greenlet==3.5.0
greenlet==3.5.1
# via sqlalchemy
grpcio==1.80.0
grpcio==1.81.1
# via
# -r requirements/torch-cuda-requirements.txt
# opentelemetry-exporter-otlp-proto-grpc
Expand All @@ -108,7 +115,7 @@ h11==0.16.0
# -r requirements/torch-cuda-requirements.txt
# httpcore
# uvicorn
hf-xet==1.5.0
hf-xet==1.5.1
# via
# -r requirements/torch-cuda-requirements.txt
# huggingface-hub
Expand All @@ -122,13 +129,13 @@ httpx==0.28.1
# huggingface-hub
# openai
# weasel
huggingface-hub==1.15.0
huggingface-hub==1.19.0
# via
# -r requirements/torch-cuda-requirements.txt
# sentence-transformers
# tokenizers
# transformers
idna==3.15
idna==3.18
# via
# -r requirements/torch-cuda-requirements.txt
# anyio
Expand All @@ -143,7 +150,7 @@ jinja2==3.1.6
# -r requirements/torch-cuda-requirements.txt
# spacy
# torch
jiter==0.14.0
jiter==0.15.0
# via openai
jmespath==1.1.0
# via
Expand Down Expand Up @@ -178,6 +185,10 @@ murmurhash==1.0.15
# preshed
# spacy
# thinc
narwhals==2.22.1
# via
# -r requirements/torch-cuda-requirements.txt
# scikit-learn
networkx==3.6.1
# via
# -r requirements/torch-cuda-requirements.txt
Expand All @@ -195,43 +206,47 @@ numpy==1.26.2
# thinc
# torchvision
# transformers
nvidia-cublas-cu12==12.8.4.1
nvidia-cublas==13.1.1.3
# via
# nvidia-cudnn-cu12
# nvidia-cusolver-cu12
# nvidia-cudnn-cu13
# nvidia-cusolver
# torch
nvidia-cuda-cupti-cu12==12.8.90
# via torch
nvidia-cuda-nvrtc-cu12==12.8.93
# via torch
nvidia-cuda-runtime-cu12==12.8.90
# via torch
nvidia-cudnn-cu12==9.10.2.21
nvidia-cuda-cupti==13.0.85
# via cuda-toolkit
nvidia-cuda-nvrtc==13.0.88
# via
# cuda-toolkit
# nvidia-cublas
nvidia-cuda-runtime==13.0.96
# via cuda-toolkit
nvidia-cudnn-cu13==9.20.0.48
# via torch
nvidia-cufft-cu12==11.3.3.83
nvidia-cufft==12.0.0.61
# via cuda-toolkit
nvidia-cufile==1.15.1.6
# via cuda-toolkit
nvidia-curand==10.4.0.35
# via cuda-toolkit
nvidia-cusolver==12.0.4.66
# via cuda-toolkit
nvidia-cusparse==12.6.3.3
# via
# cuda-toolkit
# nvidia-cusolver
nvidia-cusparselt-cu13==0.8.1
# via torch
nvidia-cufile-cu12==1.13.1.3
nvidia-nccl-cu13==2.29.7
# via torch
nvidia-curand-cu12==10.3.9.90
# via torch
nvidia-cusolver-cu12==11.7.3.90
# via torch
nvidia-cusparse-cu12==12.5.8.93
nvidia-nvjitlink==13.0.88
# via
# nvidia-cusolver-cu12
# torch
nvidia-cusparselt-cu12==0.7.1
# via torch
nvidia-nccl-cu12==2.27.3
# via torch
nvidia-nvjitlink-cu12==12.8.93
# via
# nvidia-cufft-cu12
# nvidia-cusolver-cu12
# nvidia-cusparse-cu12
# torch
nvidia-nvtx-cu12==12.8.90
# cuda-toolkit
# nvidia-cufft
# nvidia-cusolver
# nvidia-cusparse
nvidia-nvshmem-cu13==3.4.5
# via torch
nvidia-nvtx==13.0.85
# via cuda-toolkit
openai==1.97.1
# via -r requirements/gpu-requirements.in
opentelemetry-api==1.39.1
Expand Down Expand Up @@ -290,6 +305,7 @@ packaging==26.2
# -r requirements/torch-cuda-requirements.txt
# huggingface-hub
# opentelemetry-instrumentation
# scikit-optimize
# spacy
# thinc
# transformers
Expand Down Expand Up @@ -372,24 +388,24 @@ s3transfer==0.13.1
# via
# -r requirements/torch-cuda-requirements.txt
# boto3
safetensors==0.7.0
safetensors==0.8.0
# via
# -r requirements/torch-cuda-requirements.txt
# transformers
scikit-learn==1.5.2
scikit-learn==1.9.0
# via
# -r requirements/torch-cuda-requirements.txt
# scikit-optimize
# sentence-transformers
scikit-optimize==0.9.0
scikit-optimize==0.10.2
# via -r requirements/torch-cuda-requirements.txt
scipy==1.16.3
# via
# -r requirements/torch-cuda-requirements.txt
# scikit-learn
# scikit-optimize
# sentence-transformers
sentence-transformers==5.3.0
sentence-transformers==5.5.1
# via -r requirements/gpu-requirements.in
shellingham==1.5.4
# via
Expand All @@ -416,7 +432,7 @@ srsly==2.5.3
# spacy
# thinc
# weasel
starlette==0.49.3
starlette==1.3.1
# via
# -r requirements/torch-cuda-requirements.txt
# fastapi
Expand All @@ -436,26 +452,26 @@ tokenizers==0.22.2
# via
# -r requirements/torch-cuda-requirements.txt
# transformers
torch==2.8.0
torch==2.12.0
# via
# -r requirements/torch-cuda-requirements.txt
# sentence-transformers
# torchvision
torchvision==0.23.0
torchvision==0.27.0
# via -r requirements/gpu-requirements.in
tqdm==4.67.3
tqdm==4.68.2
# via
# -r requirements/torch-cuda-requirements.txt
# huggingface-hub
# openai
# sentence-transformers
# spacy
# transformers
transformers==5.5.0
transformers==5.12.1
# via
# -r requirements/torch-cuda-requirements.txt
# sentence-transformers
triton==3.4.0
triton==3.7.0
# via torch
typer==0.25.1
# via
Expand Down Expand Up @@ -486,6 +502,7 @@ typing-extensions==4.15.0
typing-inspection==0.4.2
# via
# -r requirements/torch-cuda-requirements.txt
# fastapi
# pydantic
urllib3==2.7.0
# via
Expand All @@ -507,7 +524,7 @@ wrapt==1.17.3
# -r requirements/torch-cuda-requirements.txt
# opentelemetry-instrumentation
# smart-open
zipp==3.23.1
zipp==4.1.0
# via
# -r requirements/torch-cuda-requirements.txt
# importlib-metadata
Expand Down
2 changes: 1 addition & 1 deletion gpu.Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
ARG PARENT_IMAGE=kernai/refinery-parent-images:v2.6.0-torch-cuda
ARG PARENT_IMAGE=kernai/refinery-parent-images:v3.1.0-torch-cuda

FROM ${PARENT_IMAGE} AS builder

Expand Down
Loading