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

RuntimeError: CUBLAS error: CUBLAS_STATUS_EXECUTION_FAILED[BUG] #30

Closed
zhangxliang opened this issue Dec 30, 2021 · 5 comments
Closed
Labels
bug Something isn't working

Comments

@zhangxliang
Copy link

Describe the bug

用的CPM-2提供的docker镜像,出现该bug,其他环境没有尝试。

Minimal steps to reproduce

Expected behavior

运行examples下面的generate以及quick start中的代码都报错

Screenshots

image

Environment:

image
image

@a710128
Copy link
Collaborator

a710128 commented Dec 30, 2021

请反馈一下:使用的GPU型号,BMInf版本号,CUDA版本

@zhangxliang
Copy link
Author

zhangxliang commented Dec 30, 2021

请反馈一下:使用的GPU型号,BMInf版本号,CUDA版本

GPU: GeForce RTX 3090, BMInf: 1.0.0, CUDA: 11.1

@a710128
Copy link
Collaborator

a710128 commented Dec 30, 2021

试一下不用docker直接运行呢

@zhangxliang
Copy link
Author

试一下不用docker直接运行呢

不用docker运行不会出现类似错误,具体原因不明,可能是docker中pytorch与cuda和显卡不兼容

@a710128
Copy link
Collaborator

a710128 commented Dec 31, 2021

可能是由于docker内的cuda版本与系统环境中的不一致

@a710128 a710128 added the bug Something isn't working label Dec 31, 2021
@a710128 a710128 closed this as completed Jan 7, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working
Projects
None yet
Development

No branches or pull requests

2 participants