欢迎光临
我们一直在努力

Mujoco足式机器人强化学习训练01(MuJoCo PlayGround)

Mujoco足式机器人强化学习训练01

Hello! 大家好,这个系列将会记录本人在mujoco playground进行足式机器人训练中遇到的各种问题和个人理解,如出现错误,欢迎大家指出和讨论!!! 在这里插入图片描述

注:本人电脑环境 ubuntu:20.04 GPU :4070(8GB)(其实8GB的显存根本不够用,mujoco与isaac gym不太一样,它是使用jax进行GPU训练,第一轮次需要进行长时间编译,本人在训练过程中发现,同时训练2048个env就已经需要20GB的显存了) python:3.11

MuJoCo PlayGround环境部署

  • 利用anaconda创建虚拟环境
  • conda create -n mujoco-rl python=3.11

  • 在你想要存放代码的文件夹中git clone playground的仓库
  • git clone git@github.com:google-deepmind/mujoco_playground.git && cd mujoco_playground

  • 安装uv,可以理解为并行安装工具,可加快安装速度
  • pip install uv

  • 安装jax(CUDA),用于后期并行训练
  • uv pip install -U "jax[cuda12]"
    python -c "import jax; print(jax.default_backend())" # 打印信息为gpu

  • 安装playground
  • cd ../
    uv pip install -e ".[all]"
    python -c "import mujoco_playground" # 开始下载MuJoCo Menagerie库

    在安装playground的时候可能会被墙,需要科学上网,去官网自行下载“mujoco-3.4.1.dev854365689-cp311-cp311-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl“,笔者在撰写本文的时候,这个whl文件在清华镜像源中还没有更新,此处需要注意!!!

    6.如果后期想要使用torch进行训练(rsl_rl),后期可能还需要安装tensorboard,自行根据报错提示安装即可

    至此完成mujoco playground的所有安装步骤(相较于在jetson nano上配置yolov5,此处十分简单🐶)

    mujoco playground 镜像链接:link.

    参考文献 [1]: https://zhuanlan.zhihu.com/p/1913023899534354252 [2]: https://blog.csdn.net/gitblog_01136/article/details/147063792

    赞(0)
    未经允许不得转载:171主机测评 » Mujoco足式机器人强化学习训练01(MuJoCo PlayGround)
    分享到: 更多 (0)

    评论 抢沙发

    • 昵称 (必填)
    • 邮箱 (必填)
    • 网址