@@ -79,6 +79,7 @@ For more information about OpenRL, please refer to the [documentation](https://o
7979- [ Welcome to OpenRL] ( #welcome-to-openrl )
8080- [ Outline] ( #outline )
8181- [ Installation] ( #installation )
82+ - [ Use Docker] ( #use-docker )
8283- [ Quick Start] ( #quick-start )
8384- [ Gallery] ( #gallery )
8485- [ Projects Using OpenRL] ( #projects-using-openrl )
@@ -119,6 +120,35 @@ After installation, users can check the version of OpenRL through command line:
119120openrl --version
120121```
121122
123+ ## Use Docker
124+
125+ OpenRL currently provides Docker images with and without GPU support.
126+ If the user's computer does not have an NVIDIA GPU, they can obtain an image without the GPU plugin using the following command:
127+ ``` bash
128+ sudo docker pull openrllab/openrl-cpu
129+ ```
130+
131+ If the user wants to accelerate training with a GPU, they can obtain it using the following command:
132+ ``` bash
133+ sudo docker pull openrllab/openrl
134+ ```
135+
136+ After successfully pulling the image, users can run OpenRL's Docker image using the following commands:
137+ ``` bash
138+ # Without GPU acceleration
139+ sudo docker run -it openrllab/openrl-cpu
140+ # With GPU acceleration
141+ sudo docker run -it --gpus all --net host openrllab/openrl
142+ ```
143+
144+ Once inside the Docker container, users can check OpenRL's version and then run test cases using these commands:
145+ ``` bash
146+ # Check OpenRL version in Docker container
147+ openrl --version
148+ # Run test case
149+ openrl --mode train --env CartPole-v1
150+ ```
151+
122152## Quick Start
123153
124154OpenRL provides a simple and easy-to-use interface for beginners in reinforcement learning.
0 commit comments