All-In-One/Dockerfile.gpu at master · samimideksa/All-In-One · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
FROM nvidia/cuda:9.0-cudnn7-runtime-ubuntu16.04

RUN apt-get update
RUN apt-get install -y software-properties-common
RUN add-apt-repository ppa:deadsnakes/ppa -y
RUN apt-get update && apt-get install -y --no-install-recommends \
        git \
        build-essential \
        python3.6 \
        python3.6-dev \
        python3-pip \
        python-setuptools \
        cmake \
        wget \
        curl \
        libsm6 \
        libxext6 \
        libxrender-dev


RUN python3.6 -m pip install -U pip
RUN wget --timestamping
RUN python3.6 -m pip install -r requirements-gpu.txt

COPY . /All-In-One

WORKDIR /All-In-One

EXPOSE 50051

RUN cd grpc && python3.6 -m grpc_tools.protoc -I. --python_out=. --grpc_python_out=. all_in_one.proto

CMD ["python3.6", "grpc/allinone_server.py"]