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

Use CUDA for FaceAnalysis #3

Open
henryruhs opened this issue Nov 4, 2024 · 3 comments
Open

Use CUDA for FaceAnalysis #3

henryruhs opened this issue Nov 4, 2024 · 3 comments

Comments

@henryruhs
Copy link

henryruhs commented Nov 4, 2024

The insightface dependency has to go in my opinion as this causes lot of issues like:

libstdc++.so.6: version `GLIBCXX_3.4.32' not found

For the moment, I suggest to use cuda to speed up training:

FaceAnalysis(device='cuda')

and for inswapper just:

providers = ['CUDAExecutionProvider']
@henryruhs henryruhs changed the title Use cuda for FaceAnalysis Use CUDA for FaceAnalysis Nov 4, 2024
@somanchiu
Copy link
Owner

Based on my testing in both Windows and Linux, insightface has no issues. Do you use a Python virtual environment?

As for FaceAnalysis, CUDA has been supported in the code for both training and inference.

@henryruhs henryruhs reopened this Nov 10, 2024
@henryruhs
Copy link
Author

I couldn't find a related commit, so I opened this issue because it's missing in some areas.

@somanchiu
Copy link
Owner

It has always been supported. By default, FaceAnalysis will use CUDA.

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