LLaMA Factory 训练模型未检测到CUDA环境解决

遇上这个问题说明pytorch版本没有适配好

1.cmd查看GPU版本：

nvidia-smi

我是12.6 你通过 版本参考链接 查找你对应版本

2.执行命令

3.执行完成 验证是否成功

python
import torch
print(torch.cuda.is_available())  # 是否可以用gpu False不能，True可以
print(torch.cuda.device_count())  # gpu数量， 0就是没有，1就是检测到了

说明成功了！