uieGitHub

安装cuda

1 获取版本
我的是 CUDA Toolkit 11.7.1 (August 2022), Versioned Online Documentation
为了防止后期版本不对应，我这里小心谨慎安装了August对应的月份。


C:\Users\89735>nvidia-smi
Mon Dec 19 21:31:28 2022
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 517.00       Driver Version: 517.00       CUDA Version: 11.7     |
|-------------------------------+----------------------+----------------------+
| GPU  Name            TCC/WDDM | Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  NVIDIA GeForce ... WDDM  | 00000000:01:00.0 Off |                  N/A |
| N/A    0C    P0    23W /  N/A |      0MiB /  6144MiB |      0%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|  No running processes found                                                 |
+-----------------------------------------------------------------------------+

2 下载对应版本

3 运行命令行查看

这一步我没做，直接运行命令行也弹出了他的信息，所以先略过。
4 安装cudnn

也是成功出现两个pass

完结~！

下面进入使用飞浆部分。

UIE的使用

UIE信息抽取

pycharm使用

依次运行这俩

pip install --upgrade paddlenlp
 pip show paddlenlp

然后发现运行官方demo报错

# -*- coding:utf-8 -*-f
from pprint import pprint
from paddlenlp import Taskflow

schema = ['时间', '选手', '赛事名称'] # Define the schema for entity extraction
ie = Taskflow('information_extraction', schema=schema, model='uie-base')
ie_en = Taskflow('information_extraction', schema=schema, model='uie-base-en')
pprint(ie("2月8日上午北京冬奥会自由式滑雪女子大跳台决赛中中国选手谷爱凌以188.25分获得金牌！")) # Better print results using pprint