From 2588da0e27bd5193498cff31b136031814b2761a Mon Sep 17 00:00:00 2001
From: Joseph Young <130690493+yousecjoe@users.noreply.github.com>
Date: Wed, 20 Mar 2024 18:33:34 -0400
Subject: [PATCH] Update PyTorch wheel source to CUDA 11.8

Modified the Dockerfile to install PyTorch, torchvision, and torchaudio from a CUDA 11.8 specific wheel URL. This ensures compatibility with the CUDA version in our environment and potentially improves performance and stability for GPU-accelerated operations.
---
 Dockerfile-cuda | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Dockerfile-cuda b/Dockerfile-cuda
index b8064e5c5..782d9745e 100644
--- a/Dockerfile-cuda
+++ b/Dockerfile-cuda
@@ -86,7 +86,7 @@ RUN apk update && \
     apk del /var/cache/apk/*.tbz2
 
 COPY ./backend/requirements.txt ./requirements.txt
-RUN pip3 install torch torchvision torchaudio --no-cache-dir && \
+RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 --no-cache-dir && \
     pip3 install -r requirements.txt --no-cache-dir
 
 # copy built frontend files