introlab / rtabmap

RTAB-Map library and standalone application
https://introlab.github.io/rtabmap
Other
2.84k stars 786 forks source link

SuperPoint with no GPU #1176

Open takahiko-hasegawa-hdjp opened 11 months ago

takahiko-hasegawa-hdjp commented 11 months ago

HI,

Im trying to use SuperPoint without GPU in a docker environement. Is there a way to do that?

thanks in advance

matlabbe commented 11 months ago

Yes, you may start from an existing docker image containing pytorch already installed like we did in this one: https://github.com/introlab/rtabmap/blob/master/docker/frontiers2022/Dockerfile

Similarly if you need the ros version: https://github.com/introlab/rtabmap_ros/blob/master/docker/noetic/superpoint/Dockerfile

You can launch rtabmap with SuperPoint/Cuda parameter set to false to use the CPU.