req

Dockerfile

@@ -29,7 +29,8 @@ RUN chown -R user:user $WORKDIR
 RUN chmod -R 777 $WORKDIR
 
 COPY requirements.txt $WORKDIR/requirements.txt
-RUN cat $WORKDIR/requirements.txt | xargs -n 1 -L 1 pip install --no-cache-dir
+RUN pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu113
+RUN pip install --no-cache-dir --upgrade -r $WORKDIR/requirements.txt
 RUN pip install ninja
 
 COPY . .

requirements.txt

@@ -33,10 +33,6 @@ safetensors==0.2.7
 timm==0.6.12
 transformers==4.25.1
 yapf==0.32.0
---extra-index-url https://download.pytorch.org/whl
-torch==1.11.0+cu113 
-torchvision==0.12.0+cu113
-torchaudio==0.11.0+cu113
 numpy==1.23
 git+https://github.com/facebookresearch/detectron2.git
 opencv-python