Repository navigation
不支持多卡运行问题 #48
Description
Activity
--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.
似乎不支持多卡？可以直接指定具体的device，例如：
--device cuda:0可以直接指定具体的device，例如：
--device cuda:0麻烦问下如何指定多张Gpu？
可以直接指定具体的device，例如：
--device cuda:0不行的，它这个device必须指定多卡，不然就报错
--divice改成auto后，启动服务报错