add: cuda 12.6

Author: thr3a

cuda12.4-torch/Dockerfile

@@ -8,7 +8,7 @@ ENV TZ=Asia/Tokyo
 ENV PYTHONUNBUFFERED=1
 ENV PIP_NO_CACHE_DIR=on
 ENV PYTHONDONTWRITEBYTECODE=1
-ENV UV_PROJECT_ENVIRONMENT="/usr/local/"
+ENV UV_PROJECT_ENVIRONMENT="/usr/"
 
 RUN apt-key adv --keyserver keyserver.ubuntu.com --recv f23c5a6cf475977595c89f51ba6932366a755776 \
  && echo "deb http://ppa.launchpadcontent.net/deadsnakes/ppa/ubuntu jammy main" > /etc/apt/sources.list.d/python.list \
@@ -19,8 +19,8 @@ RUN apt-get update \
  && ln -nfs /usr/bin/python${PYTHON_VERSION} /usr/bin/python \
  && ln -nfs /usr/bin/python${PYTHON_VERSION} /usr/bin/python3 \
  && rm -rf /var/lib/apt/lists/* \
- && curl -sSL https://bootstrap.pypa.io/get-pip.py | python -
-RUN pip install uv
+ && curl -sSL https://bootstrap.pypa.io/get-pip.py | python - \
+ && pip install uv
 RUN pip install torch torchvision --index-url https://download.pytorch.org/whl/cu124
 
 WORKDIR /app

cuda12.4-torch/README.md

@@ -27,7 +27,7 @@ services:
 Dockerfile
 
 ```dockerfile
-FROM ghcr.io/thr3a/cuda12.4-torch:latest
+FROM cuda12.4-torch:latest
 
 WORKDIR /app
 # COPY ./requirements.txt ./

cuda12.6-torch/Dockerfile

@@ -0,0 +1,26 @@
+FROM --platform=linux/x86_64 nvcr.io/nvidia/cuda:12.6.3-cudnn-devel-ubuntu24.04
+
+ARG PYTHON_VERSION=3.13
+ARG PACKAGES="git curl ca-certificates vim wget unzip build-essential cmake jq"
+
+ENV DEBIAN_FRONTEND=noninteractive
+ENV TZ=Asia/Tokyo
+ENV PYTHONUNBUFFERED=1
+ENV PIP_NO_CACHE_DIR=on
+ENV PYTHONDONTWRITEBYTECODE=1
+ENV UV_PROJECT_ENVIRONMENT="/usr/"
+
+RUN apt-key adv --keyserver keyserver.ubuntu.com --recv f23c5a6cf475977595c89f51ba6932366a755776 \
+ && echo "deb http://ppa.launchpadcontent.net/deadsnakes/ppa/ubuntu jammy main" > /etc/apt/sources.list.d/python.list \
+ && echo "deb-src http://ppa.launchpad.net/deadsnakes/ppa/ubuntu jammy main" >> /etc/apt/sources.list.d/python.list
+
+RUN apt-get update \
+ && apt-get install -y --no-install-recommends ${PACKAGES} python${PYTHON_VERSION} \
+ && ln -nfs /usr/bin/python${PYTHON_VERSION} /usr/bin/python \
+ && ln -nfs /usr/bin/python${PYTHON_VERSION} /usr/bin/python3 \
+ && rm -rf /var/lib/apt/lists/* \
+ && curl -sSL https://bootstrap.pypa.io/get-pip.py | python - \
+ && pip install uv
+RUN pip install torch torchvision --index-url https://download.pytorch.org/whl/cu126
+
+WORKDIR /app

cuda12.6-torch/README.md

@@ -0,0 +1,39 @@
+docker-compose.yml
+
+```yml
+services:
+  app:
+    build:
+      context: .
+      dockerfile: Dockerfile
+    image: peft-workspace:latest
+    stop_grace_period: 0s
+    ipc: host
+    tty: true
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              device_ids: ['1']
+              capabilities: [gpu]
+    volumes:
+      - "./:/app"
+      - "./cache:/root/.cache"
+    # command: sleep infinity
+    command: python train_rinna.py
+```
+
+Dockerfile
+
+```dockerfile
+FROM thr3a/cuda12.6-torch:latest
+
+WORKDIR /app
+# COPY ./requirements.txt ./
+# RUN pip install -r requirements.txt
+
+COPY pyproject.toml ./
+COPY uv.lock ./
+RUN uv sync --frozen --no-cache
+```