search

Zeyi-Lin / LLM-Finetune

大语言模型微调，Qwen2、GLM4指令微调
207 stars 33 forks source link

微调时进度条不动 #3

Open liweizheng0528 opened 3 months ago

liweizheng0528 commented 3 months ago

请问这个是什么原因 1722539119761

hellocrystal commented 1 month ago

File "/ft/train.py", line 168, in
response = predict(messages, model, tokenizer) File "/ft/train.py", line 70, in predict model_inputs = tokenizer([text], return_tensors="pt").to(device) File "/root/miniconda3/lib/python3.8/site-packages/transformers/tokenization_utils_base.py", line 803, in to self.data = {k: v.to(device=device) for k, v in self.data.items() if v is not None} File "/root/miniconda3/lib/python3.8/site-packages/transformers/tokenization_utils_base.py", line 803, in self.data = {k: v.to(device=device) for k, v in self.data.items() if v is not None} File "/root/miniconda3/lib/python3.8/site-packages/torch/cuda/init.py", line 314, in _lazy_init torch._C._cuda_init() The NVIDIA driver on your system is too old (found version 11060). Please update your GPU driver by downloading and installing a new version from the URL: http://www.nvidia.com/Download/index.aspx Alternatively, go to: https://pytorch.org to install a PyTorch version that has been compiled with your version of the CUDA driver.