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

Docker hotfix #162

Merged
merged 7 commits into from
Nov 16, 2018
Merged

Docker hotfix #162

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
e854dc3
added Dockerfile
Oct 31, 2018
0aa8b4b
intructions for installation with docker
Oct 31, 2018
9c2d487
INSTALL.md: added note to inform nvidia-docker only works with Linux
Nov 12, 2018
b314279
dockerfile nccl2 hotfix
Nov 16, 2018
8264e8c
INSTALL.md update
Nov 16, 2018
f0a76c6
Merge remote-tracking branch 'upstream/master' into docker-hotfix
Nov 16, 2018
c88e97a
removes explicit install of NCCL2 as it is included in the CUDA image
Nov 16, 2018
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
6 changes: 3 additions & 3 deletions INSTALL.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -56,12 +56,12 @@ python setup.py build develop
# MACOSX_DEPLOYMENT_TARGET=10.9 CC=clang CXX=clang++ python setup.py build develop
```

### Option 2: Docker Image (Requires CUDA, Linux Only)
### Option 2: Docker Image (Requires CUDA, Linux only)

Build image with defaults (`CUDA_VERSION=9.0`, `CUDNN_VERSION=7`):
Build image with defaults (`CUDA=9.0`, `CUDNN=7`):

nvidia-docker build -t maskrcnn-benchmark docker/

Build image with other CUDA and CUDNN versions:

nvidia-docker build -t --build-arg CUDA_VERSION=9.2 --build-arg CUDNN_VERSION=7 maskrcnn-benchmark docker/
nvidia-docker build -t --build-arg CUDA=9.2 --build-arg CUDNN=7 maskrcnn-benchmark docker/
9 changes: 4 additions & 5 deletions docker/Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,12 +1,11 @@
ARG CUDA_VERSION=9.0
ARG CUDNN_VERSION=7
ARG CUDA="9.0"
ARG CUDNN="7"

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

# install basics
RUN apt-get update -y \
&& apt-get install -y git curl ca-certificates bzip2 cmake tree htop bmon iotop g++ \
&& apt-get install -y --no-install-recommends libnccl2=2.3.5-2+cuda${CUDA_VERSION}
&& apt-get install -y git curl ca-certificates bzip2 cmake tree htop bmon iotop g++

# Install Miniconda
RUN curl -so /miniconda.sh https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh \
Expand Down