Skip to content

Commit 121c50a

Browse files
authored
Point GR00T workflow docs at the host checkpoint path and add the Blackwell CUDA reqs (#1134)
## Summary Doc only. Point GR00T workflow docs at the host checkpoint path and add the Blackwell CUDA reqs. Fix https://nvbugspro.nvidia.com/bug/6652391, https://nvbugspro.nvidia.com/bug/6658396
1 parent 39d2f4d commit 121c50a

7 files changed

Lines changed: 39 additions & 27 deletions

File tree

docs/pages/example_workflows/locomanipulation/step_4_policy_training.rst

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -109,6 +109,13 @@ We post-train the GR00T N1.6 policy on the task.
109109
The GR00T N1.6 policy has 3 billion parameters so post-training is an expensive operation.
110110
We provide one post-training option, 8 GPUs with 48GB memory, to achieve the best quality.
111111

112+
.. note::
113+
114+
Blackwell GPUs with compute capability ``sm_120`` require CUDA 12.8 or newer. The
115+
`official GR00T documentation <https://github.com/NVIDIA/Isaac-GR00T/blob/e29d8fc50b0e4745120ae3fb72447986fe638aa6/README.md?plain=1#L102>`_
116+
specifies CUDA 12.8 and ``pytorch-cu128`` for RTX 5090 systems. Please refer to the
117+
documentation for the latest requirements.
118+
112119
Training takes approximately 4-8 hours on 8x L40s GPUs.
113120

114121
Compute Requirements:

docs/pages/example_workflows/locomanipulation/step_5_evaluation.rst

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -50,25 +50,24 @@ submodule pinned at commit ``e29d8fc``; populate it with
5050
checked out. Then, from the repository root in that terminal, launch the server and
5151
leave it running:
5252

53-
.. todo::
54-
55-
The ``submodules/Isaac-GR00T`` submodule will be removed after the policy
56-
config refactor. After that, users will be expected to set up a separate
57-
GR00T repository checkout themselves and launch the server from there.
58-
5953
.. note::
6054

6155
Blackwell GPUs with compute capability ``sm_120`` require CUDA 12.8 or newer. The
6256
`official GR00T documentation <https://github.com/NVIDIA/Isaac-GR00T/blob/e29d8fc50b0e4745120ae3fb72447986fe638aa6/README.md?plain=1#L102>`_
6357
specifies CUDA 12.8 and ``pytorch-cu128`` for RTX 5090 systems. Please refer to the
6458
documentation for the latest requirements.
6559

60+
.. note::
61+
62+
``$MODELS_DIR`` exists inside the container. On the host, the checkpoint is
63+
under the models directory passed to ``./docker/run_docker.sh -m`` (default ``$HOME/models``).
64+
6665
.. code-block:: bash
6766
6867
cd submodules/Isaac-GR00T
6968
uv run python gr00t/eval/run_gr00t_server.py \
7069
--modality-config-path ../../isaaclab_arena_gr00t/embodiments/g1/g1_sim_wbc_data_config.py \
71-
--model-path /models/isaaclab_arena/locomanipulation_tutorial/checkpoint-20000 \
70+
--model-path $HOME/models/isaaclab_arena/locomanipulation_tutorial/checkpoint-20000 \
7271
--embodiment-tag NEW_EMBODIMENT \
7372
--device cuda --host 127.0.0.1 --port 5555
7473

docs/pages/example_workflows/sequential_static_manipulation/step_4_policy_training.rst

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -120,6 +120,13 @@ We provide two post-training options:
120120
* Best Quality: 8 GPUs with 48GB memory
121121
* Low Hardware Requirements: 1 GPU with 24GB memory
122122

123+
.. note::
124+
125+
Blackwell GPUs with compute capability ``sm_120`` require CUDA 12.8 or newer. The
126+
`official GR00T documentation <https://github.com/NVIDIA/Isaac-GR00T/blob/e29d8fc50b0e4745120ae3fb72447986fe638aa6/README.md?plain=1#L102>`_
127+
specifies CUDA 12.8 and ``pytorch-cu128`` for RTX 5090 systems. Please refer to the
128+
documentation for the latest requirements.
129+
123130

124131
.. tab-set::
125132

docs/pages/example_workflows/sequential_static_manipulation/step_5_evaluation.rst

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -50,25 +50,24 @@ submodule pinned at commit ``e29d8fc``; populate it with
5050
checked out. Then, from the repository root in that terminal, launch the server and
5151
leave it running:
5252

53-
.. todo::
54-
55-
The ``submodules/Isaac-GR00T`` submodule will be removed after the policy
56-
config refactor. After that, users will be expected to set up a separate
57-
GR00T repository checkout themselves and launch the server from there.
58-
5953
.. note::
6054

6155
Blackwell GPUs with compute capability ``sm_120`` require CUDA 12.8 or newer. The
6256
`official GR00T documentation <https://github.com/NVIDIA/Isaac-GR00T/blob/e29d8fc50b0e4745120ae3fb72447986fe638aa6/README.md?plain=1#L102>`_
6357
specifies CUDA 12.8 and ``pytorch-cu128`` for RTX 5090 systems. Please refer to the
6458
documentation for the latest requirements.
6559

60+
.. note::
61+
62+
``$MODELS_DIR`` exists inside the container. On the host, the checkpoint is
63+
under the models directory passed to ``./docker/run_docker.sh -m`` (default ``$HOME/models``).
64+
6665
.. code-block:: bash
6766
6867
cd submodules/Isaac-GR00T
6968
uv run python gr00t/eval/run_gr00t_server.py \
7069
--modality-config-path ../../isaaclab_arena_gr00t/embodiments/gr1/gr1_arms_only_data_config.py \
71-
--model-path /models/isaaclab_arena/sequential_static_manipulation_tutorial/checkpoint-20000 \
70+
--model-path $HOME/models/isaaclab_arena/sequential_static_manipulation_tutorial/checkpoint-20000 \
7271
--embodiment-tag GR1 \
7372
--device cuda --host 127.0.0.1 --port 5555
7473

docs/pages/example_workflows/static_manipulation/step_4_policy_training.rst

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -113,6 +113,13 @@ We provide two post-training options:
113113
* Best Quality: 8 GPUs with 48GB memory
114114
* Low Hardware Requirements: 1 GPU with 24GB memory
115115

116+
.. note::
117+
118+
Blackwell GPUs with compute capability ``sm_120`` require CUDA 12.8 or newer. The
119+
`official GR00T documentation <https://github.com/NVIDIA/Isaac-GR00T/blob/e29d8fc50b0e4745120ae3fb72447986fe638aa6/README.md?plain=1#L102>`_
120+
specifies CUDA 12.8 and ``pytorch-cu128`` for RTX 5090 systems. Please refer to the
121+
documentation for the latest requirements.
122+
116123

117124
.. tab-set::
118125

docs/pages/example_workflows/static_manipulation/step_5_evaluation.rst

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -45,25 +45,24 @@ submodule pinned at commit ``e29d8fc``; populate it with
4545
checked out. Then, from the repository root in that terminal, launch the server and
4646
leave it running:
4747

48-
.. todo::
49-
50-
The ``submodules/Isaac-GR00T`` submodule will be removed after the policy
51-
config refactor. After that, users will be expected to set up a separate
52-
GR00T repository checkout themselves and launch the server from there.
53-
5448
.. note::
5549

5650
Blackwell GPUs with compute capability ``sm_120`` require CUDA 12.8 or newer. The
5751
`official GR00T documentation <https://github.com/NVIDIA/Isaac-GR00T/blob/e29d8fc50b0e4745120ae3fb72447986fe638aa6/README.md?plain=1#L102>`_
5852
specifies CUDA 12.8 and ``pytorch-cu128`` for RTX 5090 systems. Please refer to the
5953
documentation for the latest requirements.
6054

55+
.. note::
56+
57+
``$MODELS_DIR`` exists inside the container. On the host, the checkpoint is
58+
under the models directory passed to ``./docker/run_docker.sh -m`` (default ``$HOME/models``).
59+
6160
.. code-block:: bash
6261
6362
cd submodules/Isaac-GR00T
6463
uv run python gr00t/eval/run_gr00t_server.py \
6564
--modality-config-path ../../isaaclab_arena_gr00t/embodiments/gr1/gr1_arms_only_data_config.py \
66-
--model-path /models/isaaclab_arena/static_manipulation_tutorial/checkpoint-20000 \
65+
--model-path $HOME/models/isaaclab_arena/static_manipulation_tutorial/checkpoint-20000 \
6766
--embodiment-tag GR1 \
6867
--device cuda --host 127.0.0.1 --port 5555
6968

docs/pages/quickstart/running_a_real_policy/gr00t.rst

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -27,12 +27,6 @@ submodule pinned at commit ``e29d8fc``. Populate it if needed:
2727

2828
Then start the server from the repository root in a separate shell:
2929

30-
.. todo::
31-
32-
The ``submodules/Isaac-GR00T`` submodule will be removed after the policy
33-
config refactor. After that, users will set up a separate GR00T repository
34-
checkout and launch the server from there.
35-
3630
.. code-block:: bash
3731
3832
cd submodules/Isaac-GR00T

0 commit comments

Comments
 (0)