运行train.py时报错

Clouditera / SecGPT

SecGPT网络安全大模型

Apache License 2.0

1.91k stars 248 forks source link

运行train.py时报错 #54

Open GOOD-N-LCM opened 5 months ago

GOOD-N-LCM commented 5 months ago

我在准备好自己的数据集之后基于现有的secgpt进行二次训练再运行 train.py 时报错：RuntimeError: CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling cublasCreate(handle) 请问这是什么原因导致的？当初训练secgpt遇到同样的错误您是如何解决的。期待您的解答谢谢！

skyvast404 commented 4 months ago

解决了吗兄弟

GOOD-N-LCM commented 4 months ago

解决了吗兄弟

没得，后来使用llama-factory训了