mirror of
https://github.com/LibreTranslate/LibreTranslate.git
synced 2024-12-22 23:26:31 +00:00
Merge pull request #427 from TheTakylo/main
Fix docker cuda image build
This commit is contained in:
commit
4852234636
2 changed files with 3 additions and 3 deletions
2
.github/workflows/publish-docker.yml
vendored
2
.github/workflows/publish-docker.yml
vendored
|
@ -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
|
||||
|
|
|
@ -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
|
||||
|
||||
|
|
Loading…
Reference in a new issue