从远程ssh登录运行mlagents_envs UnityEnvironment

如何解决从远程ssh登录运行mlagents_envs UnityEnvironment

我有一个脚本，其中构建了一个mlagents_envs.environments.UnityEnvironment，当我从ubuntu机器（具有GUI）上启动的终端会话运行脚本时，该脚本可以成功启动并运行。而且，如果我进入计算机，则可以从最初在计算机本地创建的tmux会话中运行这些脚本。但是，如果我尝试从通过远程ssh连接创建的终端会话中运行脚本，则在尝试创建UnityEnvironment时脚本将挂起。它只是说：

Found path: <path_to_unity_executable>

并最终超时。

我试图用虚拟显示器运行脚本，但仍然无法正常工作。具体来说，我已经尝试过：

$ xvfb-run --auto-servernum --server-args='-screen 1 640x480x24:64' python3 python_script.py -batchmode

$ xvfb-run --auto-servernum --server-args='-screen 1 640x480x24:64' python3 python_script.py

我已经尝试过此处的说明：https://github.com/Unity-Technologies/ml-agents/blob/master/docs/Training-on-Amazon-Web-Service.md

有人遇到过这个问题吗？你有什么建议吗？

解决方法

解决方案最终变得非常简单。我只需要在运行脚本之前指定正确的设备即可。

America/Fort_Nelson

如果还有其他人遇到此问题，则可能还需要在服务器和客户端上的ssh设置中都启用X11转发。我不确定100％。