scott/rocm-chatterbox-whisper
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Revert "Switch to ROCm 6.4 + Python 3.11 + PyTorch 2.9.1"
All checks were successful
Build ROCm Image / build (push) Successful in 45s
Details

Browse Source 
This reverts commit eb6a39d292.
This commit is contained in:
scott
2026-04-06 18:05:43 -04:00
parent eb6a39d292
commit 66445fad84
2 changed files with 12 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
Dockerfile.rocm
View File

@@ -1,4 +1,4 @@
FROM rocm/dev-ubuntu-22.04:6.4 FROM rocm/dev-ubuntu-22.04:6.1
ENV DEBIAN_FRONTEND=noninteractive \ ENV DEBIAN_FRONTEND=noninteractive \
PYTHONDONTWRITEBYTECODE=1 \ PYTHONDONTWRITEBYTECODE=1 \
@@ -7,37 +7,27 @@ ENV DEBIAN_FRONTEND=noninteractive \
PIP_NO_CACHE_DIR=1 PIP_NO_CACHE_DIR=1
RUN apt-get update && apt-get install -y --no-install-recommends \ RUN apt-get update && apt-get install -y --no-install-recommends \
software-properties-common \ python3 \
&& add-apt-repository ppa:deadsnakes/ppa \ python3-pip \
&& apt-get update && apt-get install -y --no-install-recommends \ python3-dev \
python3.11 \
python3.11-dev \
python3.11-distutils \
git \ git \
ffmpeg \ ffmpeg \
libsndfile1 \ libsndfile1 \
&& rm -rf /var/lib/apt/lists/* && rm -rf /var/lib/apt/lists/*
# Install pip for Python 3.11
RUN curl -sS https://bootstrap.pypa.io/get-pip.py | python3.11
# Make python3.11 the default python3
RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.11 1 \
&& update-alternatives --install /usr/bin/python python /usr/bin/python3.11 1
WORKDIR /app WORKDIR /app
# Step 1: Install ROCm-compatible PyTorch stack first. # Step 1: Install ROCm-compatible PyTorch stack first.
# This must happen before anything else to prevent pip from pulling CPU wheels. # This must happen before anything else to prevent pip from pulling CPU wheels.
COPY requirements-rocm-init.txt . COPY requirements-rocm-init.txt .
RUN python3 -m pip install -r requirements-rocm-init.txt RUN pip3 install -r requirements-rocm-init.txt
# Step 2: Install remaining dependencies (pinned to avoid overwriting torch). # Step 2: Install remaining dependencies (pinned to avoid overwriting torch).
COPY requirements-rocm.txt . COPY requirements-rocm.txt .
RUN python3 -m pip install -r requirements-rocm.txt RUN pip3 install -r requirements-rocm.txt
# Step 3: Install chatterbox with --no-deps so pip cannot replace ROCm torch. # Step 3: Install chatterbox with --no-deps so pip cannot replace ROCm torch.
RUN python3 -m pip install --no-deps chatterbox-tts RUN pip3 install --no-deps chatterbox-tts
# Application source # Application source
COPY engine.py config.py wyoming_handler.py wyoming_voices.py main.py ./ COPY engine.py config.py wyoming_handler.py wyoming_voices.py main.py ./

10
requirements-rocm-init.txt
View File

@@ -1,5 +1,5 @@
--index-url https://download.pytorch.org/whl/rocm6.4 --index-url https://download.pytorch.org/whl/rocm6.1
torch==2.9.1 torch==2.5.1
torchaudio==2.9.1 torchaudio==2.5.1
torchvision==0.24.1 torchvision==0.20.1
pytorch_triton_rocm==3.5.1 pytorch_triton_rocm==3.1.0