File tree Expand file tree Collapse file tree 1 file changed +54
-1
lines changed
Expand file tree Collapse file tree 1 file changed +54
-1
lines changed Original file line number Diff line number Diff line change 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+
You can’t perform that action at this time.
0 commit comments