Coder/whishper
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

improve torch installation and bum buildx

Browse Source
This commit is contained in:
pluja
2023-10-27 07:53:06 +02:00
parent de54570735
commit 16d4541a47
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/docker-transcription-api-gpu.yml vendored
View File

@@ -17,7 +17,7 @@ jobs:
uses: docker/setup-qemu-action@v3
- name: Set up Docker Buildx
uses: docker/setup-buildx-action@v2
uses: docker/setup-buildx-action@v3
- name: Login to DockerHub
uses: docker/login-action@v3

2
transcription-api/Dockerfile.gpu
View File

@@ -19,7 +19,7 @@ RUN ln -s -f /usr/bin/python${PYTHON_VERSION} /usr/bin/python3 && \
COPY requirements.txt requirements.txt
RUN pip3 install -r requirements.txt \
&& pip3 install python-multipart \
&& pip3 install torch==1.13.0+cu117 -f https://download.pytorch.org/whl/torch
&& pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
WORKDIR /app