Dockerfile.gpu

FROM tensorflow/tensorflow:latest-gpu

MAINTAINER RLgraph <dev@rlgraph.org>

# Install dependencies
RUN apt-get update && apt-get install -y --no-install-recommends \
    build-essential \
    curl \
    libfreetype6-dev \
    libpng12-dev \
    libzmq3-dev \
    pkg-config \
    python \
    python-dev \
    rsync \
    software-properties-common \
    cmake \
    swig \
    zlib1g-dev \
    unzip \
    git \
    && \
apt-get clean && \
    rm -rf /var/lib/apt/lists/*

# Install pip
RUN curl -O https://bootstrap.pypa.io/get-pip.py && \
    python get-pip.py && \
    rm get-pip.py
 
WORKDIR /rl-benchmark
ADD . .

RUN rm -rf *.png _cmd_*.txt

# Add dependencies
RUN pip install -r requirements_benchmark_gpu.txt

RUN pip install -e .

RUN [ -d benchmarks ] || mkdir benchmarks && rm -rf benchmarks/*

ENTRYPOINT ["python", "scripts/benchmark_gym.py"]