Update Dockerfile

Dockerfile

@@ -1,5 +1,5 @@
-# Use the CUDA 11.8 base image
-FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
+# STEP 1: Change this line to a CUDA 11.7.1 base image
+FROM nvidia/cuda:11.7.1-devel-ubuntu22.04
 
 # Set up the environment
 ENV DEBIAN_FRONTEND=noninteractive
@@ -19,9 +19,8 @@ ENV HF_HOME=/app/cache
 # Create the cache directory and give it open permissions
 RUN mkdir -p /app/cache && chmod 777 /app/cache
 
-# ⭐⭐⭐ ADD THIS BLOCK ⭐⭐⭐
-# Install PyTorch and related libraries for CUDA 11.8 separately
-RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
+# STEP 2: Change this line to install PyTorch for CUDA 11.7
+RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117
 
 # Copy and install the rest of the Python dependencies
 COPY requirements.txt .