Skip to content

Commit eb57419

Browse files
committed
Specified that a GPU can be used within the docker container for faster inference
1 parent 208d4be commit eb57419

File tree

1 file changed

+19
-5
lines changed

1 file changed

+19
-5
lines changed

docs/demos/manipulation.md

Lines changed: 19 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -87,23 +87,37 @@ manipulation techniques.
8787
8888
#### 1. Setting up the demo
8989
90-
1. Set up docker as outlined in the [docker setup guide](../setup/setup_docker.md). During the setup, build the docker image with all dependencies (i.e., use the `--build-arg DEPENDENCIES=all_groups` argument)
91-
and configure communication between the container and the host ([link](../setup/setup_docker.md#2-set-up-communications-between-docker-and-host-optional)).
90+
1. Set up docker as outlined in the [docker setup guide](../setup/setup_docker.md). During the setup, build the docker image with all dependencies (i.e., use the `--build-arg DEPENDENCIES=all_groups` argument)
91+
and configure communication between the container and the host ([link](../setup/setup_docker.md#2-set-up-communications-between-docker-and-host-optional)).
9292
93-
2. On the host machine, download the latest binary release for the Robotic Arm Demo:
93+
2. On the host machine, download the latest binary release for the Robotic Arm Demo:
9494
9595
```shell
9696
./scripts/download_demo.sh manipulation
9797
```
9898
99-
3. (Inside the container shell) Download additional ROS 2 dependencies:
99+
3. Run the docker container (if not already running):
100+
101+
```shell
102+
docker run --net=host --ipc=host --pid=host -e ROS_DOMAIN_ID=$ROS_DOMAIN_ID -it rai:jazzy # or rai:humble
103+
```
104+
105+
!!! tip "NVIDIA GPU acceleration"
106+
107+
If the [NVIDIA Container Toolkit](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html) is set up on your host machine, you can use the GPU within the RAI docker container for faster inference by adding the `--gpus all` option:
108+
109+
```shell
110+
docker run --net=host --ipc=host --pid=host -e ROS_DOMAIN_ID=$ROS_DOMAIN_ID --gpus all -it rai:jazzy # or rai:humble
111+
```
112+
113+
4. (Inside the container shell) Download additional ROS 2 dependencies:
100114
101115
```shell
102116
vcs import < demos.repos
103117
rosdep install --from-paths src/examples/rai-manipulation-demo/ros2_ws/src --ignore-src -r -y
104118
```
105119
106-
4. (Inside the container shell) Build the ROS 2 workspace:
120+
5. (Inside the container shell) Build the ROS 2 workspace:
107121
108122
```shell
109123
colcon build --symlink-install

0 commit comments

Comments
 (0)