Hardware-accelerated PyTorch on Ubuntu 20.04 for deploying yolo_ros with ROS 2 (Humble) and Python 3.8.
To run the docker with the nvidia dependencies, it is neccesary to install the NVIDIA Container Toolkit and to configure Docker to recognize the container toolkit.
To run the container:
docker run -it --rm --ipc=host --runtime=nvidia --network=host --privileged <image> bash
At ~/ros2_ws
, it is installed the yolo_ros
package with its dependecies. Using bash, this workspace and the humble workspace will be sourced. To launch the package, you can launch:
ros2 launch yolo_bringup yolov11.launch.py
Python Version | Dockerhub Image Name | Size |
---|---|---|
Python3.8 |
agonzc34/yolo-ros:humble-l4t-r32.7.1 |
16.8 GB |
note: Make sure to run the container on the latest L4T host system (r32.7.1). Running on older JetPack releases (e.g. r32.6.1) can cause driver issues, since L4T drivers are passed into the container
This image based on the Jetson_Ubuntu20_Images