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:
parent
e4e2f7f1ed
commit
50d0f2fa1e
24
Dockerfile
24
Dockerfile
@ -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
|
||||
|
@ -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]
|
Loading…
Reference in New Issue
Block a user