NVIDIA Isaac Gym安装与使用
NVIDIA做的Isaac Gym,个人理解就是一个类似于openai的Gym,不过把环境的模拟这个部分扔到了GPU上进行,这样可以提升RL训练的速度。
官网:https://developer.nvidia.com/isaac-gym
一篇blog:https://syncedreview.com/2021/09/01/deepmind-podracer-tpu-based-rl-frameworks-deliver-exceptional-performance-at-low-cost-95/
安装
官网里面选择立即加入,然后下载对应的Isaac Gym package,解压之后打开docs/index.html
,按照里面的Install in an existing Python environment
步骤进行安装即可。主要步骤就一步,即在python/
目录下执行pip install -e .
。
在 python/examples/ 目录下执行,python joint_monkey.py
来检验是否安装正确。
可能出现的问题
如果出现了
ImportError: libpython3.7m.so.1.0: cannot open shared object file: No such file or directory
的问题,可以尝试
sudo apt install libpython3.7
# 或者
export LD_LIBRARY_PATH=/path/to/libpython/directory
# 或者对于使用conda的用户
export LD_LIBRARY_PATH=/path/to/conda/envs/your_env/lib
# 自行替换 /path/to 这个路径
需要注意的是安装时需要cuda version大于等于11.4,不然会出现运行Isaac Gym时无法使用GPU的问题,比如出现:`[Warning] [carb.gym.plugin] Failed to create a valid PhysX CUDA Context Manager. Falling back to CPU.`。
如果是不带显示器输出的服务器运行isaacgym的话,可能会存在无法display的问题,类似于这种错误: ``` [Error] [carb.windowing-glfw.plugin] GLFW initialization failed. [Error] [carb.windowing-glfw.plugin] GLFW window creation failed! [Error] [carb.gym.plugin] Failed to create Window in CreateGymViewerInternal *** Failed to create viewer ```
可以参照这两个solutions (我没有试过) :
- https://forums.developer.nvidia.com/t/windowglfw-failed-create-window/122934
- https://forums.developer.nvidia.com/t/when-i-ran-the-examples/186377
如果渲染界面出现黑屏并且提示Isaac gym无响应(Isaac gym no response)的话:我只在部分机器上见过这种情况,查看`joint_monkey.py`的代码吼发现是267行`draw_viewer`函数卡住了,后来无意中搜到了解决方法: https://forums.developer.nvidia.com/t/running-examples-hangs/189609 ,可能是程序选错了device
export VK_ICD_FILENAMES=/etc/vulkan/icd.d/nvidia_icd.json
即可
使用
若要利用Isaac Gym跑起一个agent,可以查看这个仓库:https://github.com/NVIDIA-Omniverse/IsaacGymEnvs
未完待续...