文件"/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版本。
文件"/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版本。