Add files via upload

Dockerfile

@@ -1,17 +1,19 @@
-FROM ubuntu:22.04
+FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
 ENV HOST 0.0.0.0
 RUN apt-get update && apt-get upgrade -y \
     && apt-get install -y git build-essential portaudio19-dev \
     python3 python3-pip gcc wget \
+    ocl-icd-opencl-dev opencl-headers clinfo \
+    libclblast-dev libopenblas-dev libaio-dev \
+    && mkdir -p /etc/OpenCL/vendors && echo "libnvidia-opencl.so.1" > /etc/OpenCL/vendors/nvidia.icd \
     && ln -s /usr/bin/python3 /usr/bin/python
 
-RUN mkdir -p /app && mkdir -p /app/pip_cache
-ENV PIP_CACHE_DIR=/app/pip_cache
-
 WORKDIR /app
 COPY . .
-RUN pip install --no-cache-dir --no-deps -r system/requirements/requirements_standalone.txt --cache-dir=/app/pip_cache
+ENV CUDA_DOCKER_ARCH=all
+RUN pip install --no-cache-dir --no-deps -r system/requirements/requirements_docker.txt && \
+    pip install --no-cache-dir deepspeed
 
-EXPOSE 7851
+EXPOSE 7851 7852
 RUN chmod +x launch.sh
 ENTRYPOINT ["sh", "-c", "./launch.sh"]

docker-compose.yml

@@ -1,13 +1,21 @@
 version: '3.8'
 services:
   alltalk-tts:
-    image: erew123/alltalk_tts:cpu
+    image: alltalk:v1-9c
     restart: unless-stopped
     ports:
       - "7851:7851"
+      - "7852:7852"
     volumes:
       - ./outputs:/app/outputs/
       - ./models:/app/models/
       - ./voices:/app/voices/
       - ./finetune/put-voice-samples-in-here:/app/finetune/put-voice-samples-in-here
       - ./dockerconfig.json:/app/confignew.json
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities: [ gpu ]