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

Compile Issue #135

Closed
RodneyHe opened this issue Aug 4, 2024 · 1 comment
Closed

Compile Issue #135

RodneyHe opened this issue Aug 4, 2024 · 1 comment

Comments

@RodneyHe
Copy link

RodneyHe commented Aug 4, 2024

Hi, is there a method can compile SAM 2 in the nvidia driver with old version? My project is deployed in cluster, the gpu driver version is fixed (gpu driver version 470.74; cuda driver version 11.4; cuda runtime version 11.6), I tried to update the cuda runtime version to 12.1 and install corresponding pytorch 2.3, but it doesn't work, still got issue of old nvidia driver version.

@ronghanghu
Copy link
Contributor

Hi, this is likely due to a failure in SAM 2 CUDA extension building. We have recently made the CUDA extension step optional (in #155) as a workaround to this problem.

You can pull the latest code and reinstall via

# run the line below inside the SAM 2 repo
git pull;
pip uninstall -y SAM-2;
rm -f sam2/*.so;
pip install -e ".[demo]"

which allows using SAM 2 without CUDA extension (the results should stay the same in most cases, see INSTALL.md for details).

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

2 participants