OpenBMB/BMInf

[BUG]RuntimeError: Library cublasLt is not initialized

hxl523 opened this issue · 0 comments

Describe the bug

运行路径https://github.com/OpenBMB/BMInf 的demo时候,出现RuntimeError: Library cublasLt is not initialized错误

Minimal steps to reproduce

import bminf #成功导入
cpm2 = bminf.models.CPM2() #成功定义
cpm2.fill_blank('好') #报错 RuntimeError: Library cublasLt is not initialized
Expected behavior

Screenshots

image
image

Environment:

NVIDIA-SMI 465.19.01
Driver Version: 465.19.01
NVIDIA A40
CUDA Version: 11.3
Memory:45634MiB