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

单机多卡跑gradio推理时,报CUDA的错误 #6

Closed
chelovek21 opened this issue Jun 9, 2023 · 0 comments
Closed

单机多卡跑gradio推理时,报CUDA的错误 #6

chelovek21 opened this issue Jun 9, 2023 · 0 comments
Labels
question Further information is requested

Comments

@chelovek21
Copy link

问题

单机多卡【4*RTX4090 24G】gradio推理时,加载模型可以成功,但问答时报CUDA的错误,能提供一下您运行的基础环境或可能的解决思路吗?谢谢
RuntimeError: CUDA error: device-side assert triggered
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.

基础环境

  • [1] Linux python3.10.8
  • [2] pytorch=1.13.1
  • [3] transformers=4.29.1
  • [4] accelerate=0.20.3
  • [5] peft=0.3.0
@chelovek21 chelovek21 added the question Further information is requested label Jun 9, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
question Further information is requested
Projects
None yet
Development

No branches or pull requests

1 participant