ros2_pytorch_cuda
Model runner
Enables running PyTorch models on ROS2 using CUDA devices in a containerized environment.
ROS2 PyTorch template node - running PyTorch C++ models in ROS2
7 stars
0 watching
1 forks
Language: C++
last commit: over 3 years ago
Linked from 1 awesome list
fkromer/awesome-ros2Related projects:
| Repository | Description | Stars |
|---|---|---|
 klintan/ros2_pytorch | A ROS2 node template to run PyTorch models in C++ | 39 |
 randl/mobilenetv2-pytorch | An implementation of MobileNetV2 in PyTorch for image classification tasks. | 271 |
| randl/shufflenetv2-pytorch | An implementation of a lightweight convolutional neural network architecture for mobile devices | 191 |
 torch/cutorch | Provides a CUDA backend for the PyTorch deep learning framework | 337 |
 pytorchbearer/torchbearer | A PyTorch model fitting library designed to simplify the process of training deep learning models. | 636 |
 ros/urdfdom_headers | Provides core data structure headers for describing robot models and interactions | 29 |
 4uiiurz1/pytorch-res2net | Implementations of deep learning architectures using PyTorch for image classification tasks on various datasets. | 112 |
 rentruewang/koila | A lightweight wrapper around PyTorch to prevent CUDA out-of-memory errors and optimize model execution | 1,823 |
 amdegroot/pytorch-containers | A collection of PyTorch implementations of Torch Table Layers to simplify the transition from old Torch architectures. | 89 |
 clcarwin/convert_torch_to_pytorch | Converts PyTorch models to their Torch equivalents. | 541 |
| ros/urdf | Provides tools and data structures for working with 3D models in robotics | 63 |
 fxia22/pointnet.pytorch | This is an implementation of the PointNet algorithm in PyTorch for 3D point cloud classification and segmentation tasks. | 2,175 |
 prlz77/resnext.pytorch | Reproduces ResNet-V3 with PyTorch for computer vision tasks | 511 |
 viking-sudo-rm/stacknn | This project provides differentiable stacks and queues in PyTorch to facilitate the integration of data structures into models. | 142 |
 wizaron/pytorch-cpp-inference | A repository providing tools and examples to serve PyTorch models as C++ inference applications. | 226 |