search

rpautrat / SuperPoint

Efficient neural feature detector and descriptor
MIT License
1.88k stars 416 forks source link

PyTorch version for the PyTorch model? #312

Closed AruniRC closed 9 months ago

AruniRC commented 10 months ago

Could you please provide the environment details for the PyTorch model, e.g. PyTorch version, CUDA/CuDNN version, etc? Thank you!

Tuhinm2002 commented 9 months ago

You can install pytorch by using the command pip install torch Then use the python code device = "cuda" if torch.cuda.is_available() else "cpu"

rpautrat commented 9 months ago

The Pytorch model is a very simple pytorch script, so pretty much any version of Pytorch/CUDA/CuDNN should work, as long as they are all compatible between them.