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.

Sign up for GitHub

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

Jump to bottom

检测不到GPU 环境完全是按照requirements.txt里面配的 #22

Open

GIANNIS-CHEN opened this issue Jun 1, 2024 · 1 comment

GIANNIS-CHEN commented Jun 1, 2024

Collaborator

ziyannchen commented Aug 27, 2024 •

edited

Loading

你好，目前我们的代码cpu运行支持得不是很好，这里报错的地方因为我们默认使用了cuda（因为比从cpu搬运到cuda的运行效率更高），可以尝试把报错倒数第二个文件里transformer.py那里.cuda删除即可，也就是直接用torch.FloatTensor()

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