Запуск RVC в Docker на CPU

git clone https://github.com/RVC-Project/Retrieval-based-Voice-Conversion-WebUI src

Пропатчить файл infer/modules/train/train.py. Подробнее…

# os.environ["CUDA_VISIBLE_DEVICES"] = hps.gpus.replace("-", ",")
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

FROM python:3.8-slim
EXPOSE 7865
WORKDIR /app
RUN apt update \
 && DEBIAN_FRONTEND=noninteractive \
    apt install -y -qq --no-install-recommends \
        aria2 \
        build-essential \
        ffmpeg \
 && apt clean
COPY src .
RUN cd /app/ \
 && pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu \
 && pip3 install --no-cache-dir -r requirements.txt

VOLUME [ "/app/assets/weights", "/app/dataset", "/app/opt" ]

RUN python3 ./tools/download_models.py

CMD ["python3", "./infer-web.py"]

services:
  rvc:
    build:
      context: .
      dockerfile: Dockerfile
    shm_size: "8GB" # https://github.com/pytorch/pytorch/issues/2244#issuecomment-318864552
    container_name: rvc
    volumes:
      - ./weights:/app/assets/weights
      - ./dataset:/app/dataset
      - ./opt:/app/opt
    ports:
      - "7865:7865"