Added NVIDIA GPU support to Docker (RVC-Project#1231)

* Added NVIDIA GPU support to Docker * feat: Added NVIDIA GPU support to Docker --------- Co-authored-by: Abdellah Derfoufi <[email protected]>
qixingxx · Sep 12, 2023 · 50d0f2f · 50d0f2f
1 parent e4e2f7f
commit 50d0f2f
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 4 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -1,16 +1,34 @@
 # syntax=docker/dockerfile:1
 
-FROM python:3.10-bullseye
+FROM nvidia/cuda:11.6.2-cudnn8-runtime-ubuntu20.04
 
 EXPOSE 7865
 
 WORKDIR /app
 
 COPY . .
 
-RUN apt update && apt install -y -qq ffmpeg aria2 && apt clean
+# Install dependenceis to add PPAs
+RUN apt-get update && \
+    apt-get install -y -qq ffmpeg aria2 && apt clean && \
+    apt-get install -y software-properties-common && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
 
-RUN pip3 install --no-cache-dir -r requirements.txt
+# Add the deadsnakes PPA to get Python 3.9
+RUN add-apt-repository ppa:deadsnakes/ppa
+
+# Install Python 3.9 and pip
+RUN apt-get update && \
+    apt-get install -y build-essential python-dev python3-dev python3.9-distutils python3.9-dev python3.9 curl && \
+    apt-get clean && \
+    update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.9 1 && \
+    curl https://bootstrap.pypa.io/get-pip.py | python3.9
+
+# Set Python 3.9 as the default
+RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.9 1
+
+RUN python3 -m pip install --no-cache-dir -r requirements.txt
 
 RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main/pretrained_v2/D40k.pth -d assets/pretrained_v2/ -o D40k.pth
 RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main/pretrained_v2/G40k.pth -d assets/pretrained_v2/ -o G40k.pth

diff --git a/docker-compose.yml b/docker-compose.yml
@@ -10,4 +10,11 @@ services:
       - ./opt:/app/opt
       # - ./dataset:/app/dataset # you can use this folder in order to provide your dataset for model training
     ports:
-      - 7865:7865
+      - 7865:7865
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: 1
+              capabilities: [gpu]