search

ghosthamlet / gpt2-ml-torch

Pytorch model for https://github.com/imcaspar/gpt2-ml
Apache License 2.0
79 stars 16 forks source link

没有显卡运行不了啊，不支持cpu运算吗 #27

Open jianchang512 opened 2 years ago

jianchang512 commented 2 years ago

AssertionError: Torch not compiled with CUDA enabled

ghosthamlet commented 2 years ago

@jianchang512 可以用CPU，不过这个gpt2模型比较大，用CPU太慢了，至少需要1080TI的GPU才比较流畅。 命令行运行加--no_gpu参数就可以：

python gpt2_ml_torch/generate.py --prompt 宇宙的意义是 --max_len 30 --n_seq 2 --no_gpu