greet001的博客

在安装GPU的时候，踩坑不少，现安装成功，特分享以记录 一、查看本机GPU版本 桌面–右键—打开NIADIA控制面板—帮助—系统信息—组件—NIVDIA.DLL 查看cuda最高可安装版本 二、官网下载与系统匹配的cuda https://www.geforce.com/hardware/technology/cuda/supported-gpus 亦可以百度搜索对应cuda下载，然后安装 如：https://developer.nvidia.com/cuda-11.3.0-download-archive

一、准备数据集 Dataset 其使用需要自定义一个类继承Dataset—》class ImdbDataset(Dataset): 其主要作用是从文件中获取数据，主要有三个方法： __init__该方法是初始化方法，用来获取所有文件的路径 __getitems__该方法是通过index索引获取到每个文件的内容（包括content,label）,此处利用tokenlize对获取到的文本了预处理（处理掉不必要的字符） __len__该方法是获取整个数据集的长度 注意： 数据集分为训练集和测试集，所以此处在处理