self-hosted/LibreTranslate
1
0
Fork 0
mirror of https://github.com/LibreTranslate/LibreTranslate.git synced 2024-12-18 08:27:03 +02:00
Code Issues Releases Activity

Merge pull request #427 from TheTakylo/main

Browse Source 
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/publish-docker.yml vendored
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

4
docker/cuda.Dockerfile
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