[THUDM/ChatGLM-6B]CUDA版本需求

请问这个项目的CUDA版本有要求吗，我用的11.3跑起来就报了这个错RuntimeError: CUDA Error: no kernel image is available for execution on the device，网上查了原因就说是CUDA版本不对，换了10.0跑起来的时候就说CUDA没法启动

bash train.sh

Environment

- OS:
- Python:
- Transformers:
- PyTorch:
- CUDA Support (`python -c "import torch; print(torch.cuda.is_available())"`) :
与requirement一致

z1968357787

你看看你的pytorch版本和cuda版本。实测cuda11.3+pytorch 1.11.0可以运行

XiaTiaoQAQ

你看你的pytorch版本和cuda版本。实测cuda11.3+pytorch 1.11.0可以运行

是在服务器上跑还是pc电脑？

ZXStudio

Linux服务器上跑的

z1968357787

cuda 11.7 运行python web_demo.py正常，但是运行ptuning下的bash train.py就报错。 RuntimeError: Library cudart is not initialized

speeder179

CUDA Error: no kernel image is available for execution on the device 我也有这个错误，CUDA Version: 10.2，torch==1.12.1+cu102 torchvision==0.13.1+cu102 torchaudio==0.12.1

lanss315425

Duplicate of #119

zhangch9

[THUDM/ChatGLM-6B]CUDA版本需求

回答