diff --git a/Dockerfile b/Dockerfile index 9aae39e..f6de520 100644 --- a/Dockerfile +++ b/Dockerfile @@ -11,21 +11,23 @@ RUN apt-get update -qq \ && apt-get clean \ && rm -rf /var/lib/apt +RUN apt-get update && apt-get upgrade --assume-yes + RUN pip install --upgrade pip COPY . . RUN if [ "$with_models" = "true" ]; then \ - # install only the dependencies first - pip install -e .; \ - # initialize the language models - if [ ! -z "$models" ]; then \ - ./install_models.py --load_only_lang_codes "$models"; \ - else \ - ./install_models.py; \ - fi \ - fi + # install only the dependencies first + pip install -e .; \ + # initialize the language models + if [ ! -z "$models" ]; then \ + ./install_models.py --load_only_lang_codes "$models"; \ + else \ + ./install_models.py; \ + fi \ + fi # Install package from source code RUN pip install . \ && pip cache purge diff --git a/docker-compose.cuda.yml b/docker-compose.cuda.yml index 5f8beaa..4535243 100644 --- a/docker-compose.cuda.yml +++ b/docker-compose.cuda.yml @@ -9,4 +9,10 @@ services: restart: unless-stopped ports: - 5000:5000 - command: --gpus all + deploy: + resources: + reservations: + devices: + - driver: nvidia + count: 1 + capabilities: [gpu] diff --git a/docker/Dockerfile.cuda b/docker/Dockerfile.cuda index 337a32d..6bd1603 100644 --- a/docker/Dockerfile.cuda +++ b/docker/Dockerfile.cuda @@ -12,7 +12,9 @@ RUN apt-get update -qq \ && apt-get clean \ && rm -rf /var/lib/apt -RUN pip3 install --upgrade pip +RUN apt-get update && apt-get upgrade --assume-yes + +RUN pip3 install --upgrade pip && apt-get remove python3-pip --assume-yes COPY . . diff --git a/requirements.txt b/requirements.txt index 9ea3949..87ad661 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,7 +3,7 @@ Flask==2.1.2 flask-swagger==0.2.14 flask-swagger-ui==3.36.0 Flask-Limiter==2.4.5.1 -waitress==2.1.1 +waitress==2.1.2 expiringdict==1.2.1 pyicu>=2.8 pycld2==0.41