vllm [Bug]: CUDA Error when print

jm81lzqq  于 3个月前  发布在  其他
关注(0)|答案(2)|浏览(53)

当前环境

The output of `python collect_env.py`

🐛 描述bug

我刚接触vllm,我想检查llama模型的输入,所以我只是打印了

然而,它引发了cuda bug。为什么?我该如何获取输入?

谢谢!

eulz3vhy

eulz3vhy1#

我认为这是由于某些CUDA设置引起的。

6ioyuze2

6ioyuze22#

如错误信息所述,请尝试使用 CUDA_LAUNCH_BLOCKING=1 再次操作。

相关问题