NVIDIA-AI-IOT / CUDA-PointPillars

A project demonstrating how to use CUDA-PointPillars to deal with cloud points data from lidar.
Apache License 2.0
502 stars 148 forks source link

how to install Export Tools Pointpillar Onnx Model #89

Closed Acuno41 closed 1 year ago

Acuno41 commented 1 year ago

Which CUDA, Pytorch and sub packages(spconv, onnx, onnx-simplifier, onnx_graphsurgeon, etc) versions should install for successful model export ?

It says Cuda 11.4 and pytorch 1.11.0, but there isn't any pytorch version for CUDA 11.4.

Is there anyone who successfully done exporting can help me please