Ray Dockerfile

Dockerfile

FROM nvidia/cuda:10.0-cudnn7-runtime as base

RUN apt-get update && apt-get install -y --no-install-recommends \
    python3-pip \
    build-essential \ 
    python3-dev \ 
    # these two are needed for tensorflow/ray examples
    libglib2.0-0 \ 
    libgtk2.0-dev \
    git \
    && rm -rf /var/lib/apt

RUN pip3 install --no-cache-dir https://download.pytorch.org/whl/cu100/torch-1.0.1.post2-cp36-cp36m-linux_x86_64.whl

RUN pip3 install --no-cache-dir setuptools
RUN pip3 install --no-cache-dir wheel

COPY requirements.txt requirements.txt
RUN pip3 install --no-cache-dir -r requirements.txt

RUN pip3 install ray[rllib, debug]
RUN pip3 install tensorflow-gpu

ENV LC_ALL=C.UTF-8
ENV LANG=C.UTF-8

WORKDIR /code

requirements.txt

torchvision
wandb
tensorboardX
lmdb
matplotlib
numpy
plotly
gym[atari]
h5py