secgpt webdemo.py运行

1.错误1:

RuntimeError: Expected one of cpu, cuda, ipu, xpu, mkldnn, opengl, opencl, ideep, hip, ve, fpga, ort, xla, lazy, vulkan, mps, meta, hpu, mtia, privateuseone device type at start of device string: auto

解决办法:将以下代码中device直接设置为”cuda"
image.png

2.错误2:
回答很慢且乱码,或是报RuntimeError: probability tensor contains either inf, nan or element < 0
image.png
image.png
在代码中加上:

os.environ['CUDA_VISIBLE_DEVICES'] = "0,1"

image.png
根据自己的gpu性能和显存情况设置os.environ[‘CUDA_VISIBLE_DEVICES’],若还是推理很慢或报显存不够的错误,首先看设置的gpu有没有被别的程序占用,若没有就多设几张卡,如“0,1,2”。
ps: secgpt是基于baichuan-13b训练的,运行baichuan时也可能出现RuntimeError: probability tensor contains either inf, nan or element < 0错误,同样可以通过设置os.environ[‘CUDA_VISIBLE_DEVICES’]解决

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值