Skip to content

Commit

Permalink
Merge branch 'main' into ss/igx_igpu_mps
Browse files Browse the repository at this point in the history
Signed-off-by: sohams <[email protected]>
  • Loading branch information
sohamm17 committed Sep 24, 2024
2 parents 39105a7 + 3407462 commit 0b7fb13
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion applications/body_pose_estimation/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ find_package(holoscan 1.0 REQUIRED CONFIG

# Download the yolov8 pose model if it doesn't exist
add_custom_command(OUTPUT "${HOLOHUB_DATA_DIR}/body_pose_estimation/yolov8l-pose.onnx"
COMMAND yolo export model=yolov8l-pose.pt format=onnx
COMMAND "PATH=${PATH}:/workspace/holohub/.local/bin" /usr/local/bin/yolo export model=yolov8l-pose.pt format=onnx
COMMAND mkdir -p "${HOLOHUB_DATA_DIR}/body_pose_estimation"
COMMAND mv yolov8l-pose.onnx "${HOLOHUB_DATA_DIR}/body_pose_estimation"
BYPRODUCTS "${HOLOHUB_DATA_DIR}/body_pose_estimation/yolov8l-pose.onnx"
Expand Down
2 changes: 1 addition & 1 deletion run
Original file line number Diff line number Diff line change
Expand Up @@ -350,7 +350,7 @@ setup() {

# Install the tensorrt development libs
# Find the newest cudart version installed
for version in 12-4 12-3 12-2 12-1 12-0 11-8 11-7 11-6 11-4
for version in 12-6 12-5 12-4 12-3 12-2 12-1 12-0 11-8 11-7 11-6 11-4
do
cuda_version=$(dpkg --status cuda-cudart-${version} 2>/dev/null | grep -Po '^Version: \K[^-]*')
# Find the version based on cudart
Expand Down

0 comments on commit 0b7fb13

Please sign in to comment.