fix

Dockerfile

 FROM nvidia/cuda:11.3.1-base-ubuntu20.04
 ENV DEBIAN_FRONTEND=noninteractive
 
-#Install Torch(chonky!)
-RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
-
 #Install core packages
 RUN apt-get update && apt-get install -y \
       libncurses5 python3 python3-pip curl git apt-utils ssh ca-certificates \
@@ -13,6 +10,9 @@ RUN apt-get update && apt-get install -y \
     update-alternatives --install /usr/bin/pip pip /usr/bin/pip3 1 && \
     pip3 install --no-cache-dir --upgrade pip
 
+#Install Torch(chonky!)
+RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
+
 #Install Python deps
 RUN pip3 install --no-cache-dir dotmap icecream sentry-sdk numpy fastapi "uvicorn[standard]" gunicorn
 RUN pip3 install omegaconf transformers einops