CUDA版本需求

[z1968357787]

Is there an existing issue for this?

• [X] I have searched the existing issues

Current Behavior

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

Expected Behavior

No response

Steps To Reproduce

bash train.sh

Environment

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

Anything else?

No response

[XiaTiaoQAQ]

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

[ZXStudio]

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

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

[z1968357787]

Linux服务器上跑的

 

变质的水果糖 @.***

 

------------------ 原始邮件 ------------------ 发件人: "Hao @.>; 发送时间: 2023年4月13日(星期四) 晚上7:21 收件人: @.>; 抄送: @.>; @.>; 主题: Re: [THUDM/ChatGLM-6B] CUDA版本需求 (Issue #587)

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

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

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>

[speeder179]

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

[lanss315425]

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

[zhangch9]

Duplicate of #119