Zeyi-Lin / LLM-Finetune

大语言模型微调,Qwen2、GLM4指令微调
174 stars 27 forks source link

cuda驱动版本太低 #4

Open hellocrystal opened 2 hours ago

hellocrystal commented 2 hours ago

文件"/ft/train.py",第168行: response = predict(messages, model, tokenizer) 文件"/ft/train.py",第70行: model_inputs = tokenizer([text], return_tensors="pt").to(device) 文件"/root/miniconda3/lib/python3.8/site-packages/transformers/tokenization_utils_base.py",第803行: self.data = {k: v.to(device=device) for k, v in self.data.items() if v is not None} 文件"/root/miniconda3/lib/python3.8/site-packages/transformers/tokenization_utils_base.py",第803行: self.data = {k: v.to(device=device) for k, v in self.data.items() if v is not None} 文件"/root/miniconda3/lib/python3.8/site-packages/torch/cuda/init.py",第314行: torch._C._cuda_init() 你系统的NVIDIA驱动程序版本太旧(发现版本11,060)。请从网址http://www.nvidia.com/Download/index.aspx下载并安装新版本,或者从https://pytorch.org安装一个已与您所安装的CUDA驱动程序版本编译的PyTorch版本。

hellocrystal commented 2 hours ago

千问3 千问LLM大模型