-
Notifications
You must be signed in to change notification settings - Fork 69
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
Is there a way to run this model on Apple M1 that doesn't have CUDA support? #50
Comments
The error occurs in the face detection package, that is used to generate the face crops for 6DRepNet. |
For me it worked to set line 17 in site-packages/sixdrepnet/regressor.py from For me that is close enough. :) |
I get the following error when I try to run the SixDRepNet() model on an image.
File "/Users/gurpreetmukker/Desktop/face_detection/face_detection/lib/python3.11/site-packages/torch/cuda/__init__.py", line 239, in _lazy_init raise AssertionError("Torch not compiled with CUDA enabled") AssertionError: Torch not compiled with CUDA enabled
Thanks
The text was updated successfully, but these errors were encountered: