daviestechlabs/kuberay-images
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: update to python 3.12.
Some checks failed
Build and Push Images / determine-version (push) Successful in 4s
Details
Build and Push Images / build (Dockerfile.ray-worker-nvidia, nvidia) (push) Failing after 21s
Details
Build and Push Images / build (Dockerfile.ray-worker-intel, intel) (push) Failing after 23s
Details
Build and Push Images / build (Dockerfile.ray-worker-strixhalo, strixhalo) (push) Failing after 19s
Details
Build and Push Images / build (Dockerfile.ray-worker-rdna2, rdna2) (push) Failing after 23s
Details
Build and Push Images / Release (push) Has been skipped
Details
Build and Push Images / Notify (push) Successful in 1s
Details

Browse Source
This commit is contained in:
Billy Davies
2026-02-09 08:52:32 -05:00
parent 64585dac7e
commit 6aad7ad38a
4 changed files with 21 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
dockerfiles/Dockerfile.ray-worker-rdna2
View File

@@ -6,13 +6,13 @@
# docker build -t git.daviestechlabs.io/daviestechlabs/ray-worker-rdna2:latest \
# -f dockerfiles/Dockerfile.ray-worker-rdna2 .
#
# Multi-stage build: Extract ROCm from vendor image, use Ray base for Python 3.11
# Multi-stage build: Extract ROCm from vendor image, use Ray base for Python 3.12
# Stage 1: ROCm libraries from AMD vendor image
FROM docker.io/rocm/pytorch:rocm6.4.4_ubuntu22.04_py3.10_pytorch_release_2.7.1 AS rocm-source
# Stage 2: Production image
FROM docker.io/rayproject/ray:2.53.0-py311 AS production
FROM docker.io/rayproject/ray:2.53.0-py312 AS production
# OCI Image Spec labels
LABEL org.opencontainers.image.title="Ray Worker - AMD RDNA 2"
@@ -56,7 +56,7 @@ COPY --from=ghcr.io/astral-sh/uv:latest /uv /usr/local/bin/uv
USER ray
# Install PyTorch with ROCm 6.2 wheels for Python 3.11 (uv is 10-100x faster)
# Install PyTorch with ROCm 6.2 wheels for Python 3.12 (uv is 10-100x faster)
RUN --mount=type=cache,target=/home/ray/.cache/uv,uid=1000,gid=1000 \
uv pip install --system \
torch==2.5.1 torchvision torchaudio \