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 mac gpu #376

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open

support mac gpu #376

wants to merge 1 commit into from

Conversation

GcsSloop
Copy link

支持 mac 操作系统的 GPU 加速功能,在可以使用 cuda 时优先使用 cuda,cuda 不可用时判断是否使用 mps 加速,否则使用 cpu 进行推理。

@GcsSloop
Copy link
Author

GcsSloop commented Feb 14, 2025

支持 mac 操作系统的 M1 - M4 系列芯片的 GPU 推理功能。

image

在 Mac 上的运行效果

iShot_2025-02-15_01.16.22.mp4

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

Successfully merging this pull request may close these issues.

1 participant