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

[email protected] compatibility with nvcr-24.11 #637

Open
LeoAtlanto opened this issue Dec 6, 2024 · 3 comments
Open

[email protected] compatibility with nvcr-24.11 #637

LeoAtlanto opened this issue Dec 6, 2024 · 3 comments

Comments

@LeoAtlanto
Copy link

Hi,

Thanks for your great work!

I'm trying to install [email protected], which happens to be freshly bumped just half an hour ago, on the latest nvidia pytorch container nvcr24.11 (nvcr.io/nvidia/pytorch:24.11-py3, CUDA 12.6, torch 2.6.0a0+df5bbc09d1.nv24.11) and get the import error below after installation:
image

There seems to be some compatibility problem with pytorch or CUDA version.

Looking forward to your solution. Thanks!

@tridao
Copy link
Collaborator

tridao commented Dec 7, 2024

v.2.2.4 should work

@LeoAtlanto
Copy link
Author

v.2.2.4 should work

Thanks for your reply. But still got the error below with pip install git+https://[email protected]/state-spaces/[email protected]:
image

@tridao
Copy link
Collaborator

tridao commented Dec 8, 2024

I see. I'll investigate

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