New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

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

Open

jianchang512 opened this issue Jan 27, 2022 · 1 comment

jianchang512 commented Jan 27, 2022

AssertionError: Torch not compiled with CUDA enabled

Owner

ghosthamlet commented Jan 27, 2022

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

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment