Skip to content

Commit 6deb208

Browse files
authored
updated the manifest to support GPU (#73)
1 parent b8d6512 commit 6deb208

File tree

3 files changed

+19
-4
lines changed

3 files changed

+19
-4
lines changed

flowstate/services/Dockerfile.service

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,10 +6,14 @@ FROM ${BASE_IMAGE}
66
# Copy the zenoh config file.
77
# TODO(Yadunund): Switch to ZENOH_CONFIG_OVERRIDE after the next jazzy sync.
88
COPY zenoh_config.json5 /opt/zenoh_config.json5
9+
# For one-shot solutions - need to have the 3d_models directory in the folder
10+
# containing the 3d models used by the solution
11+
ADD 3d_models /opt/ros/underlay/install/3d_models
912

1013
# Set environment variables
1114
ENV ZENOH_SESSION_CONFIG_URI=/opt/zenoh_config.json5
1215
ENV ZENOH_ROUTER_CHECK_ATTEMPTS=5
16+
ENV LD_LIBRARY_PATH="/usr/local/nvidia/lib64:/usr/local/nvidia/bin:${LD_LIBRARY_PATH:-}"
1317
# (Optional) If you need to override the entrypoint or command, do it here.
1418
# For example:
1519
# ENTRYPOINT ["/your/new/entrypoint.sh"]

flowstate/services/README.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -29,15 +29,15 @@ cd ~/
2929
git clone https://github.com/intrinsic-ai/sdk.git
3030
cd sdk
3131
bazel run //intrinsic/tools/inbuild -- service bundle \
32-
--manifest ~/bpc_ws/src/bpc/services/flowstate/bpc_pose_estimator.manifest.textproto \
33-
--oci_image ~/bpc_ws/src/bpc/services/flowstate/bpc_pose_estimator.tar \
34-
--output ~/bpc_ws/src/bpc/services/flowstate/bpc_pose_estimator.bundle.tar
32+
--manifest ~/bpc_ws/src/bpc/flowstate/services/bpc_pose_estimator.manifest.textproto \
33+
--oci_image ~/bpc_ws/src/bpc/flowstate/services/bpc_pose_estimator.tar \
34+
--output ~/bpc_ws/src/bpc/flowstate/services/bpc_pose_estimator.bundle.tar
3535
```
3636

3737
Finally install the Service in a running solution.
3838

3939
```bash
40-
bazel run //intrinsic/tools/inctl -- service install ~/bpc_ws/src/bpc/services/flowstate/bpc_pose_estimator.bundle.tar \
40+
bazel run //intrinsic/tools/inctl -- service install ~/bpc_ws/src/bpc/flowstate/services/bpc_pose_estimator.bundle.tar \
4141
--cluster CLUSTER_ID \
4242
--org ORG \
4343
--registry REGISTRY \

flowstate/services/bpc_pose_estimator.manifest.textproto

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,11 +15,22 @@ service_def {
1515
real_spec {
1616
image {
1717
archive_filename: "bpc_pose_estimator.tar"
18+
settings {
19+
resource_requirements {
20+
limits: { key: "nvidia.com/gpu" value: "1" }
21+
}
22+
requires_rtpc_node: true
23+
}
1824
}
1925
}
2026
sim_spec {
2127
image {
2228
archive_filename: "bpc_pose_estimator.tar"
29+
settings {
30+
resource_requirements {
31+
limits: { key: "nvidia.com/gpu" value: "1" }
32+
}
33+
}
2334
}
2435
}
2536
}

0 commit comments

Comments
 (0)