记录一下跑模型时遇到的错误:
RuntimeError: cutlassF: no kernel found to launch!
模型推理的时候出现这个问题搜了搜本以为是bfloat精度问题,但改了之后依然没用。
nvidia-smi的cuda version是12.2
nvcc -V 的cuda version是12.1
torch版本为2.1.2
最终将torch版本回退为2.1.0后成功解决问题。
记录一下跑模型时遇到的错误:
RuntimeError: cutlassF: no kernel found to launch!
模型推理的时候出现这个问题搜了搜本以为是bfloat精度问题,但改了之后依然没用。
nvidia-smi的cuda version是12.2
nvcc -V 的cuda version是12.1
torch版本为2.1.2
最终将torch版本回退为2.1.0后成功解决问题。