Skip to content
This repository was archived by the owner on Oct 31, 2023. It is now read-only.

Update Dockerfile and install.md by editing versions which is available nowdays. #1273

Open
wants to merge 2 commits into
base: main
Choose a base branch
from
Open

Update Dockerfile and install.md by editing versions which is available nowdays. #1273

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
846dd08
Update Dockerfile
sbkim052 Sep 28, 2020
9cda5a4
Update INSTALL.md
sbkim052 Sep 28, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 5 additions & 0 deletions INSTALL.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -121,6 +121,11 @@ cd maskrcnn-benchmark
python setup.py build develop
```
### Option 2: Docker Image (Requires CUDA, Linux only)
Install nvidia-docker

Edit daemon.json file (refer to below link)

https://stackoverflow.com/questions/59691207/docker-build-with-nvidia-runtime

Build image with defaults (`CUDA=9.0`, `CUDNN=7`, `FORCE_CUDA=1`):

Expand Down
35 changes: 11 additions & 24 deletions docker/Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,8 @@ ARG CUDA="9.0"
ARG CUDNN="7"

FROM nvidia/cuda:${CUDA}-cudnn${CUDNN}-devel-ubuntu16.04

CMD nvidia-smi
RUN NVIDIA_VISIBLE_DEVICES=0
RUN echo 'debconf debconf/frontend select Noninteractive' | debconf-set-selections

# install basics
Expand All @@ -11,18 +12,19 @@ RUN apt-get update -y \
&& apt-get install -y libglib2.0-0 libsm6 libxext6 libxrender-dev

# Install Miniconda
RUN curl -so /miniconda.sh https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh \
&& chmod +x /miniconda.sh \
&& /miniconda.sh -b -p /miniconda \
&& rm /miniconda.sh

RUN curl -LO https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh \
&& bash Miniconda3-latest-Linux-x86_64.sh -p /miniconda -b \
&& rm Miniconda3-latest-Linux-x86_64.sh
RUN ls
ENV PATH=/miniconda/bin:$PATH

# Create a Python 3.6 environment
RUN /miniconda/bin/conda install -y conda-build \
&& /miniconda/bin/conda create -y --name py36 python=3.6.7 \
&& /miniconda/bin/conda create -y --name py36 python=3.6.8 \
&& /miniconda/bin/conda clean -ya

RUN pip install --upgrade pip

ENV CONDA_DEFAULT_ENV=py36
ENV CONDA_PREFIX=/miniconda/envs/$CONDA_DEFAULT_ENV
ENV PATH=$CONDA_PREFIX/bin:$PATH
Expand All @@ -31,26 +33,11 @@ ENV CONDA_AUTO_UPDATE_CONDA=false
RUN conda install -y ipython
RUN pip install requests ninja yacs cython matplotlib opencv-python tqdm

# Install PyTorch 1.0 Nightly
# Install PyTorch 1.4
ARG CUDA
RUN conda install pytorch-nightly cudatoolkit=${CUDA} -c pytorch \
RUN conda install pytorch==1.4 torchvision==0.5.0 cudatoolkit=${CUDA} -c pytorch \
&& conda clean -ya

# Install TorchVision master
RUN git clone https://github.com/pytorch/vision.git \
&& cd vision \
&& python setup.py install

# install pycocotools
RUN git clone https://github.com/cocodataset/cocoapi.git \
&& cd cocoapi/PythonAPI \
&& python setup.py build_ext install

# install apex
RUN git clone https://github.com/NVIDIA/apex.git \
&& cd apex \
&& python setup.py install --cuda_ext --cpp_ext

# install PyTorch Detection
ARG FORCE_CUDA="1"
ENV FORCE_CUDA=${FORCE_CUDA}
Expand Down