diff --git a/.github/workflows/docker-transcription-api-gpu.yml b/.github/workflows/docker-transcription-api-gpu.yml
index 9564874..6c07742 100644
--- a/.github/workflows/docker-transcription-api-gpu.yml
+++ b/.github/workflows/docker-transcription-api-gpu.yml
@@ -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
diff --git a/transcription-api/Dockerfile.gpu b/transcription-api/Dockerfile.gpu
index 7e86a5d..154eb78 100644
--- a/transcription-api/Dockerfile.gpu
+++ b/transcription-api/Dockerfile.gpu
@@ -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