cublas error: CUBLAS_STATUS_NOT_SUPPORTED
huhk-sysu opened this issue · 4 comments
huhk-sysu commented
运行examples/generate.py时报错,上层调用栈是functions/gemm.py的第249行。
Line 249 in d40c6f5
使用的环境为:
cuda 10.1(cublas版本为10.1.0.63)
BMInf 0.0.4 通过clone + python setup.py install方式安装
torch 1.7.1
a710128 commented
请提供一下GPU型号。
huhk-sysu commented
请提供一下GPU型号。
目前使用的是Tesla M40,有需要我可以更换其他型号的显卡尝试
a710128 commented
请提供一下GPU型号。
目前使用的是Tesla M40,有需要我可以更换其他型号的显卡尝试
BMInf目前不支持 NVIDIA Pascal 以前的架构(例如:Kepler,Maxwell ...)
你可以参考 https://en.wikipedia.org/wiki/CUDA#GPUs_supported 来选择合适的GPU。
huhk-sysu commented
请提供一下GPU型号。
目前使用的是Tesla M40,有需要我可以更换其他型号的显卡尝试
BMInf目前不支持 NVIDIA Pascal 以前的架构(例如:Kepler,Maxwell ...)
你可以参考 https://en.wikipedia.org/wiki/CUDA#GPUs_supported 来选择合适的GPU。
谢谢,更换Tesla V100测试通过。