diff --git a/Dockerfile b/Dockerfile
index 96b674013c9f0a3548eca5545277d2aa0857bd71..be2e1612dc2f2d52190b0b7f7e57c78890311e7d 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 ab8e3648b844d900e11aff3a4db859bbd6634a5d..84e8dd8e24baa85435adbdb3b5aa8155a2f3a6f1 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 89803d4479989df96539056edaaa4c76b7cf3d09..23dd9c5d800c8d91dee461e84dde0a12595109c1 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