1
0
mirror of synced 2024-11-23 23:21:03 +01:00

Added NVIDIA GPU support to Docker (#1231)

* Added NVIDIA GPU support to Docker

* feat: Added NVIDIA GPU support to Docker

---------

Co-authored-by: Abdellah Derfoufi <mohammed-abdellah.derfoufi@capgemini.com>
This commit is contained in:
Abdellah derfoufi 2023-09-12 17:18:04 +02:00 committed by GitHub
parent e4e2f7f1ed
commit 50d0f2fa1e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 29 additions and 4 deletions

View File

@ -1,6 +1,6 @@
# syntax=docker/dockerfile:1
FROM python:3.10-bullseye
FROM nvidia/cuda:11.6.2-cudnn8-runtime-ubuntu20.04
EXPOSE 7865
@ -8,9 +8,27 @@ 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

View File

@ -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]