系列文章目录
目录
前言
本笔记本提供 MuJoCo XLA (MJX) 的入门教程,MJX 是 MuJoCo 基于 JAX 的实现,适用于 RL 训练工作负载。
需要使用具有 GPU 加速功能的 Colab 运行时。如果使用的是纯 CPU 运行时,可以通过菜单 “运行时 > 更改运行时类型 ”进行切换。
一、安装 MuJoCo、MJX 和 Brax
!pip install mujoco
!pip install mujoco_mjx
!pip install brax
#@title Check if MuJoCo installation was successful
from google.

发布评论