微调时进度条不动

[liweizheng0528]

请问这个是什么原因

[hellocrystal]

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.