torch last

Dockerfile

@@ -35,6 +35,12 @@ WORKDIR $HOME/app
 # Upgrade pip
 RUN python -m pip install --no-cache-dir --upgrade pip
 
+# Copy requirements file
+COPY --chown=user requirements.txt .
+
+# Install other Python dependencies
+RUN pip install --no-cache-dir -r requirements.txt
+
 # Install PyTorch with CUDA support first
 RUN pip install --no-cache-dir \
     torch==2.0.1+cu118 \
@@ -42,12 +48,6 @@ RUN pip install --no-cache-dir \
     torchaudio==2.0.2+cu118 \
     -f https://download.pytorch.org/whl/torch_stable.html
 
-# Copy requirements file
-COPY --chown=user requirements.txt .
-
-# Install other Python dependencies
-RUN pip install --no-cache-dir -r requirements.txt
-
 # Copy the application code
 COPY --chown=user . .