Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Support for Pytorch 2.1 and newer #1

Closed
jamesobutler opened this issue Mar 15, 2024 · 1 comment
Closed

Support for Pytorch 2.1 and newer #1

jamesobutler opened this issue Mar 15, 2024 · 1 comment
Assignees

Comments

@jamesobutler
Copy link

jamesobutler commented Mar 15, 2024

Does SegmentAny3D specifically require the Cuda 11.7 compute platform and not anything newer? Or even support for CPU only?

https://github.com/mazurowski-lab/SlicerSegmentAny3D/blob/e33d0af23ef6d6458a95074b47374a2697582209/SegmentAny3D/SegmentAny3D.py#L122-L126

I typically have SlicerTotalSegmentator installed in my Slicer version and it will pull in latest pytorch. Starting with pytorch 2.1, the available compute platforms are CUDA 11.8 and CUDA 12.1. CUDA 11.7 is not available for latest PyTorch which is why SegmentAny3D is pulling in the older pytorch 2.0.x series.

@Zaferyildiz
Copy link
Member

Thank you for the recommendation. PyTorch version condition is removed with the latest commit.

@Zaferyildiz Zaferyildiz self-assigned this Mar 20, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants