facebookresearch / rlmeta

RLMeta is a light-weight flexible framework for Distributed Reinforcement Learning Research.
MIT License
284 stars 28 forks source link

Do we have a docker file for rlmeta? #18

Open stephenzwj opened 2 years ago

stephenzwj commented 2 years ago

Hi team,

It is very difficult to install the rlmeta due to the dependent packages like moolib are not easy to be installed? Do we have a docker file for trying out rlmeta?

xiaomengy commented 2 years ago

Hi, Thanks for the suggestion. Currently we don't have that. But we can keep this issue open and we can try to prepare that.

And could you provide more details about the difficulties you have when building moolib? Maybe we can help you on that.

d3sm0 commented 2 years ago
# -*- mode: dockerfile -*-
FROM nvidia/cuda:11.6.1-cudnn8-devel-ubuntu20.04

ARG PYTHON_VERSION=3.8
ENV DEBIAN_FRONTEND=noninteractive

RUN apt-get update && apt-get install -yq \
        build-essential \
        curl \
        git \
        ninja-build \
        libglib2.0-0 \
        sudo\
        wget \
        python3-opencv\
        python3 \
        python3-pip\
        python3.8-venv\
    && rm -rf /var/lib/apt/lists/*

WORKDIR /opt
RUN wget https://github.com/Kitware/CMake/releases/download/v3.23.4/cmake-3.23.4-linux-x86_64.sh && \
    chmod +x cmake-3.23.4-linux-x86_64.sh && \
    ./cmake-3.23.4-linux-x86_64.sh --skip-license --prefix=/usr/local

ENV VIRTUAL_ENV=/opt/venv
RUN python3 -m venv $VIRTUAL_ENV
ENV PATH="$VIRTUAL_ENV/bin:$PATH"

RUN python -m pip install --upgrade pip
RUN pip install numpy torch -f https://download.pytorch.org/whl/cu116 --no-cache-dir

WORKDIR /opt
RUN git clone https://github.com/facebookresearch/rlmeta && \
    cd rlmeta && \
    git submodule sync &&  \
    git submodule update --init --recursive && \
    pip install -e .

CMD ["/bin/bash"]

This worked for me :)