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

不支持多卡运行问题 #48

Open
opsvr opened this issue Nov 6, 2023 · 4 comments
Open

不支持多卡运行问题 #48

opsvr opened this issue Nov 6, 2023 · 4 comments

Comments

@opsvr
Copy link

opsvr commented Nov 6, 2023

--divice改成auto后,启动服务报错

@qubo
Copy link

qubo commented Nov 6, 2023

--device改成auto后,启动web_demo.py服务成功,但是运行任务报错
File "/root/.cache/huggingface/modules/transformers_modules/CodeShell-7B/modeling_codeshell.py", line 401, in forward
hidden_states = residual + feed_forward_hidden_states
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cuda:1!
^CKeyboard interruption in main thread... closing server.
似乎不支持多卡?

@ruixie
Copy link
Collaborator

ruixie commented Nov 7, 2023

可以直接指定具体的device,例如:--device cuda:0

@whyPeanutbutter
Copy link

可以直接指定具体的device,例如:--device cuda:0

麻烦问下如何指定多张Gpu?

@kztyes
Copy link

kztyes commented Mar 8, 2024

可以直接指定具体的device,例如:--device cuda:0

不行的,它这个device必须指定多卡,不然就报错

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

5 participants