2019-07-08 22:51:28 +02:00
|
|
|
FROM nvcr.io/nvidia/pytorch:19.05-py3
|
2019-03-18 19:45:05 +01:00
|
|
|
|
|
|
|
# Set working directory
|
|
|
|
WORKDIR /mlperf
|
|
|
|
|
|
|
|
RUN apt-get update && apt-get install -y python3-tk python-pip git tmux htop tree
|
|
|
|
|
|
|
|
# Necessary pip packages
|
|
|
|
RUN pip install --upgrade pip
|
|
|
|
|
|
|
|
COPY requirements.txt .
|
|
|
|
RUN pip install -r requirements.txt
|
|
|
|
RUN python3 -m pip install pycocotools==2.0.0
|
|
|
|
|
|
|
|
# Copy SSD code
|
|
|
|
COPY ./setup.py .
|
|
|
|
COPY ./csrc ./csrc
|
|
|
|
RUN pip install .
|
|
|
|
|
|
|
|
COPY . .
|