update:

Author: thr3a

.github/workflows/build-x64.yml

@@ -36,7 +36,7 @@ jobs:
     steps:
       - name: Free Disk Space (Ubuntu)
         uses: jlumbroso/free-disk-space@main
-        if: contains(fromJson('["cuda11.8-torch", "cuda12.1-torch", "alma-7b-server", "iopaint", "llama-cpp-python"]'), matrix.target)
+        if: contains(fromJson('["alma-7b-server", "iopaint", "llama-cpp-python"]'), matrix.target)
         with:
           # this might remove tools that are actually needed,
           # if set to "true" but frees about 6 GB

cuda11.8-torch/Dockerfile

@@ -1,24 +1,17 @@
-FROM --platform=linux/x86_64 nvcr.io/nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
+FROM --platform=linux/x86_64 python:3.11-slim
 
-ARG PYTHON_VERSION=3.11
 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}
 
-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 -
-RUN pip install torch torchvision --index-url https://download.pytorch.org/whl/cu118
+RUN pip install -U pip
+RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
 
 WORKDIR /app