Merge pull request #49 from huangshiyu13/dev

- add docker - update nlp package

Author: huangshiyu13

.dockerignore

@@ -0,0 +1,11 @@
+__pycache__/
+logs
+.pytest_cache/
+.coverage
+.coverage.*
+.idea/
+logs/
+.pytype/
+htmlcov/
+.vscode/
+.git/

Makefile

@@ -22,4 +22,10 @@ format:
 	# Reformat using black
 	black ${PYTHON_FILES} --preview
 
-commit-checks: format lint
+commit-checks: format lint
+
+docker-cpu:
+	./scripts/build_docker.sh
+
+docker-gpu:
+	USE_GPU=True ./scripts/build_docker.sh

README.md

@@ -87,7 +87,7 @@ conda install -c openrl openrl
 
 想要修改源码的用户也可以从源码安装OpenRL:
 ```bash
-git clone git@github.com:OpenRL-Lab/openrl.git && cd openrl
+git clone https://github.com/OpenRL-Lab/openrl.git && cd openrl
 pip install -e .
 ```
 

README_en.md

@@ -109,7 +109,7 @@ conda install -c openrl openrl
 Users who want to modify the source code can also install OpenRL from the source code:
 
 ```bash
-git clone git@github.com:OpenRL-Lab/openrl.git && cd openrl
+git clone https://github.com/OpenRL-Lab/openrl.git && cd openrl
 pip install -e .
 ```
 

docker/Dockerfile

@@ -0,0 +1,23 @@
+ARG PARENT_IMAGE
+FROM $PARENT_IMAGE
+
+WORKDIR /openrl
+
+ADD setup.py setup.py
+ADD openrl openrl
+ADD README.md README.md
+
+ENV VENV /root/venv
+
+RUN \
+    pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
+
+RUN \
+    python3 -m pip install --upgrade pip --no-cache-dir && \
+    python3 -m pip install --no-cache-dir . && \
+    python3 -m pip install --no-cache-dir ".[nlp]" && \
+    rm -rf $HOME/.cache/pip
+
+ENV PATH=$VENV/bin:$PATH
+
+CMD /bin/bash

scripts/build_docker.sh

@@ -0,0 +1,24 @@
+#!/bin/bash
+
+CPU_PARENT=cnstark/pytorch:2.0.0-py3.9.12-ubuntu20.04
+GPU_PARENT=cnstark/pytorch:2.0.0-py3.9.12-cuda11.8.0-ubuntu22.04
+
+TAG=openrllab/openrl
+VERSION=$(python -c "from openrl.__init__ import __version__;print(__version__)")
+
+if [[ ${USE_GPU} == "True" ]]; then
+  PARENT=${GPU_PARENT}
+  TAG="${TAG}"
+else
+  PARENT=${CPU_PARENT}
+  TAG="${TAG}-cpu"
+fi
+
+echo "docker build --build-arg PARENT_IMAGE=${PARENT} -t ${TAG}:${VERSION} . -f docker/Dockerfile"
+docker build --build-arg PARENT_IMAGE=${PARENT} -t ${TAG}:${VERSION} . -f docker/Dockerfile
+docker tag ${TAG}:${VERSION} ${TAG}:latest
+
+if [[ ${RELEASE} == "True" ]]; then
+  docker push ${TAG}:${VERSION}
+  docker push ${TAG}:latest
+fi

setup.py

@@ -24,7 +24,7 @@
 
 def get_install_requires() -> list:
     return [
-        "setuptools>=50.0",
+        "setuptools>=67.0",
         "gymnasium",
         "click",
         "termcolor",
@@ -37,7 +37,7 @@ def get_install_requires() -> list:
         "jsonargparse",
         "imageio",
         "opencv-python",
-        "pygame"
+        "pygame",
     ]
 
 
@@ -55,7 +55,6 @@ def get_extra_requires() -> dict:
         "dev": ["build", "twine"],
         "mpe": ["pyglet==1.5.27"],
         "nlp": [
-            "stable-baselines3==1.5.1a5",
             "transformers==4.18.0",
             "datasets",
             "nltk",