panhustar

怎么配置，电脑是CPU的，没有GPU

补充报错前的错误信息 bin C:\ProgramData\anaconda3\envs\deepke\lib\site-packages\bitsandbytes\libbitsandbytes_cpu.so C:\ProgramData\anaconda3\envs\deepke\lib\site-packages\bitsandbytes\cextension.py:34: UserWarning: The installed version of bitsandbytes was compiled without GPU support. 8-bit optimizers, 8-bit multiplication, and GPU quantization are unavailable. warn("The installed version of bitsandbytes was...

File "C:\ProgramData\anaconda3\envs\deepke\lib\site-packages\torch\cuda_init.py", line 210, in _lazy_init raise AssertionError("Torch not compiled with CUDA enabled") 补充信息 我的torch版本 Name: torch Version: 1.11.0 Summary: Tensors and Dynamic neural networks in Python with strong GPU...

你好 以下是DeepKE\example\ner\few-shot\conf\train\few_shot.yaml 的配置 seed: 1 bart_name: "C://Support//work//model//bart-large" dataset_name: mit-movie device: cpu num_epochs: 20 batch_size: 3 learning_rate: 5e-5 warmup_ratio: 0.01 eval_begin_epoch: 4 src_seq_ratio: 0.8 tgt_max_len: 10 num_beams: 1 length_penalty: 1 use_prompt:...

经查询torch版本如下 ，是cpu版本，为1.11.0+cpu (deepke) C:\WINDOWS\system32>python -c "import torch; print(torch.__version__)" 1.11.0+cpu

还是报错信息如下 File "C:\ProgramData\anaconda3\envs\deepke\lib\site-packages\deepke\name_entity_re\few_shot\models\model.py", line 270, in get_prompt input_tokens = self.prompt_inputs.unsqueeze(0).expand(batch_size, -1).to(self.device) File "C:\ProgramData\anaconda3\envs\deepke\lib\site-packages\torch\cuda\__init__.py", line 210, in _lazy_init raise AssertionError("Torch not compiled with CUDA enabled") AssertionError: Torch not compiled with CUDA...

非常感谢，问题已经解决，重新安装了deepke到site-packages 就可以了