Merge pull request #427 from TheTakylo/main

Fix docker cuda image build
This commit is contained in:
Piero Toffanin 2023-04-03 10:37:55 -04:00 committed by GitHub
commit 4852234636
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 deletions

View file

@ -48,7 +48,7 @@ jobs:
uses: docker/build-push-action@v2
with:
file: ./docker/cuda.Dockerfile
platforms: linux/amd64,linux/arm64
platforms: linux/amd64
tags: ${{ steps.get-variables.outputs.gh-username-lower }}/libretranslate:${{ env.TAG }}-cuda
push: true
cache-from: type=gha

View file

@ -1,7 +1,7 @@
FROM nvidia/cuda:11.2.2-devel-ubuntu20.04
ENV ARGOS_DEVICE_TYPE cuda
ARG with_models=true
ARG with_models=false
ARG models=""
WORKDIR /app
@ -34,7 +34,7 @@ RUN if [ "$with_models" = "true" ]; then \
fi
# Install package from source code
RUN pip3 install Babel==2.11.0 && python3 scripts/compile_locales.py \
RUN pip3 install Babel==2.12.1 && python3 scripts/compile_locales.py \
&& pip3 install . \
&& pip3 cache purge