Skip to content
Discussion options

You must be logged in to vote

Hi @SNicolaJ ,

This seems like an known issue (see #2628 ) since Pascal generation GPU (the P2200 that you have) support has been dropped on PyTorch.

Downgrading to PyTorch 2.8 with CUDA 12.6 should work (uv pip install torch==2.8.0 --index-url https://download.pytorch.org/whl/cu126) but that might cause dependency issues with other sleap-nn dependencies.

Alternatively, try exporting to ONNX and run inference on CPU, although that would probably be slower since GPU inference is faster. See #2628 for instructions on that.

Hope this helps!
Tom

Replies: 1 comment

Comment options

You must be logged in to vote
0 replies
Answer selected by tom21100227
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Help!
Labels
None yet
2 participants