Skip to content

cuda驱动版本太低 #4

@hellocrystal

Description

@hellocrystal

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions