OpenBMB/BMInf

cublas error: CUBLAS_STATUS_NOT_SUPPORTED

huhk-sysu opened this issue · 4 comments

运行examples/generate.py时报错,上层调用栈是functions/gemm.py的第249行。

cublasLt.checkCublasStatus( cublasLt.cublasLtMatmul(

使用的环境为:
cuda 10.1(cublas版本为10.1.0.63)
BMInf 0.0.4 通过clone + python setup.py install方式安装
torch 1.7.1

请提供一下GPU型号。

请提供一下GPU型号。

目前使用的是Tesla M40,有需要我可以更换其他型号的显卡尝试

请提供一下GPU型号。

目前使用的是Tesla M40,有需要我可以更换其他型号的显卡尝试

BMInf目前不支持 NVIDIA Pascal 以前的架构(例如:Kepler,Maxwell ...)
你可以参考 https://en.wikipedia.org/wiki/CUDA#GPUs_supported 来选择合适的GPU。

请提供一下GPU型号。

目前使用的是Tesla M40,有需要我可以更换其他型号的显卡尝试

BMInf目前不支持 NVIDIA Pascal 以前的架构(例如:Kepler,Maxwell ...)
你可以参考 https://en.wikipedia.org/wiki/CUDA#GPUs_supported 来选择合适的GPU。

谢谢,更换Tesla V100测试通过。