BlinkDL / AI-Writer

AI 写小说,生成玄幻和言情网文等等。中文预训练生成模型。采用我的 RWKV 模型,类似 GPT-2 。AI写作。RWKV for Chinese novel generation.
Apache License 2.0
2.97k stars 518 forks source link

A卡 安装完第一次跑报错 #8

Open huige728 opened 2 years ago

huige728 commented 2 years ago

用机环境win7 64位 ,e3 1230v2+hd6570,A卡驱动及管理器是安装好的 已经解决的安装问题: 1、c++库需是2019以上 2、pip install torch onnxruntime-directml是国外的源,下载太慢,要3个多小时,几kb-十几kb,转国内源就好下载

最后跑起来出现下图问题:

Loading model for dml... Traceback (most recent call last): File "d:/ai/AI writer main/run.py" line 81, in <module> import onnxruntime as rt File "C:\Users\huige\AppData\Local\Programs\Python\Python38\lib\site packages\onnxruntime\__init__ .py" line 55, in<module> raise import_capi_exception File "C: \Users\huige\AppData\Local\Prograns\Python\Python38\lib\site-packages\onnxruntime\__init__ .py", line 23, in <module> from onnxruntime. capi._pybind_state import( File "C: \Users\huige\AppData\Local\Programs\Python\Python38\lib\site-pacages\onnxruntime\capi\_pybind_ state.py", line 32, in <module> from .onnxruntime_pybind11_state import * # noqa ImportError: DLL load failed while importing onxruntime_pybind11_state: 找不到指定的模块。 PS D:\ai\AI-Writer-main>

BlinkDL commented 2 years ago

还需要win10最新21h2版,而且A卡至少需要RX6xxx。

Sum41forever commented 1 year ago

给个配置参考把,我也是用的A卡 RX580 win1022H2 能跑起来 image