From c04686eadca035412fa7ba87b26f3cc3ad8b73b8 Mon Sep 17 00:00:00 2001
From: iggotsul <igorgotsu@gmail.com>
Date: Thu, 10 Feb 2022 09:10:54 +0200
Subject: [PATCH] modified reqs
---
Dockerfile | 6 +++++-
apt.txt | 1 +
requirements.txt | 9 +++------
3 files changed, 9 insertions(+), 7 deletions(-)
diff --git a/Dockerfile b/Dockerfile
index 96b6740..be2e161 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -35,7 +35,7 @@ RUN wget -nv -O miniconda.sh https://repo.anaconda.com/miniconda/Miniconda3-py37
&& bash miniconda.sh -b -p ${CONDA_DIR} \
&& . ${CONDA_DIR}/etc/profile.d/conda.sh \
&& rm -rf miniconda.sh \
- && conda install pytorch==1.5.1 torchvision==0.6.1 cudatoolkit=10.1 -c pytorch \
+ && conda install pytorch==1.8.0 torchvision==0.9.0 cudatoolkit=10.1 -c pytorch \
&& conda clean -a -y
ENV FORCE_CUDA="1"
@@ -46,9 +46,13 @@ COPY --chown=1001:1001 requirements.txt requirements.txt
RUN pip install --no-cache-dir -r requirements.txt
RUN pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.8.0/index.html
+#RUN apt-get install jupyter-notebook
+
# (optional)
# (either add below as dependency in requirements.txt or uncomment below line)
# RUN pip install -U git+https://github.com/open-mmlab/mmdetection.git@v2.3.0
# Copy user files
COPY --chown=1001:1001 . ${HOME_DIR}
+
+#CMD while true; do sleep 1; done
diff --git a/apt.txt b/apt.txt
index ab8e364..84e8dd8 100644
--- a/apt.txt
+++ b/apt.txt
@@ -7,3 +7,4 @@ zlib1g-dev
libgl1-mesa-glx
libglib2.0-0
git
+jupyter-notebook
diff --git a/requirements.txt b/requirements.txt
index 89803d4..23dd9c5 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -9,9 +9,6 @@ aicrowd-api
opencv-python
pyyaml==5.1
--f https://download.pytorch.org/whl/cu101/torch_stable.html
-torch==1.8.0+cu101
-torchvision==0.9.0+cu101
-
--f https://dl.fbaipublicfiles.com/detectron2/wheels/cu101/torch1.8/index.html
-detectron2==0.6
+timeout_decorator
+git+https://github.com/AIcrowd/coco.git#subdirectory=PythonAPI
+git+https://github.com/open-mmlab/mmdetection.git@v2.3.0
--
GitLab