Update docker

AlexeySrus · May 28, 2022 · 6600038 · 6600038
1 parent 246448d
commit 6600038
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -10,3 +10,12 @@ RUN apt-get update && \
 RUN pip3 install --upgrade pip
 
 
+RUN pip3 install torch==1.10 torchvision==0.11 torchaudio==0.10 --extra-index-url https://download.pytorch.org/whl/cu113
+
+RUN pip3 install openmim
+RUN mim install mmdet
+
+COPY . /nenets_walsuses
+WORKDIR /nenets_walsuses/
+ENV PYTHONPATH=/nenets_walsuses
+