Skip to content

Commit 51a96e4

Browse files
update readme
1 parent a5e877f commit 51a96e4

File tree

1 file changed

+54
-1
lines changed

1 file changed

+54
-1
lines changed

README.md

Lines changed: 54 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1,54 @@
1-
# EpMineEnv
1+
# EpMineEnv
2+
3+
## 环境配置
4+
5+
### 使用本地环境(有显示器)
6+
7+
```
8+
conda create -p python3.8 mine_env
9+
conda activate mine_env
10+
pip install mlagents-envs gym opencv-python==4.5.5.64
11+
```
12+
13+
根据系统配置和[官方文档](https://pytorch.org/get-started/locally/)安装pytorch。
14+
15+
根据[Stable-Baseline3](https://stable-baselines3.readthedocs.io/en/master/guide/install.html)官方文档安装强化学习框架。
16+
17+
### 使用远程服务器(无显示器)
18+
19+
为了使仿真环境支持服务器端的图像渲染,建议使用docker的方式。需要安装`nvidia-docker`,详细的安装过程建议参考[https://github.com/ValerioB88/ml-agents-visual-observations-docker-GPU](https://github.com/ValerioB88/ml-agents-visual-observations-docker-GPU)
20+
21+
我们也提供了已经配置好环境的镜像,上述过程配置完成后,可以直接拉取镜像
22+
23+
```
24+
docker pull haoranlee/mine_visual_gpu:v0
25+
```
26+
27+
使用该镜像时,可以将本地的代码和仿真环境所在文件夹映射到docker容器中
28+
```
29+
docker run -it -e DISPLAY=:0 -v /your/workspace/path:/work/code --network host --runtime=nvidia --privileged --entrypoint /bin/bash haoranlee/mine_visual_gpu:v0
30+
```
31+
32+
需要注意的是,在有显示器的机器上使用上述docker方式,仍然会弹出程序可视化窗口。
33+
34+
35+
### 关闭可视化界面
36+
mlagents-envs提供了`no-graphics`仿真模式,但是在该模式下图像不会被正常渲染。
37+
这里我们提供了一种通过修改mlagents-envs源码的方式,让它们支持不显示可视化窗口。
38+
具体的,找到当前python环境的库安装路径,并找到`site-packages/mlagents_envs/environment.py`,将第272行
39+
`args += ["-nographcis", "-batchmode"]` 修改为 `args += ["-batchmode"]`
40+
41+
然后再代码(`envs/SingleAgent/mine_toy.py`)中 `no_graph = True`
42+
43+
需要注意的是,上述修改方式虽然支持关闭可视化窗口,但是在服务器(无显示)端仅修改上述代码而不适用docker的情况下,仍然不能正常渲染图像。
44+
45+
***警告***:上述代码涉及修改mlagents-envs源码,请谨慎使用。
46+
47+
## 仿真环境下载
48+
[release](https://github.com/DRL-CASIA/EpMineEnv/releases/tag/v1.0)标签下,下载系统对应的仿真环境,解压到`envs/SingleAgent/`路径下,并检查`envs/SingleAgent/mine_toy.py`中的`file_name`路径是否正确。
49+
50+
在Linux系统下,需要赋予仿真环境`drl.x86_64`文件可执行权限,具体如下
51+
```
52+
chmod +x drl.x86_64
53+
```
54+

0 commit comments

Comments
 (0)