小编Ege*_*Ege的帖子

当我尝试使用 Huggingface 模型之一进行情感分析时，出现以下错误：

RuntimeError: CUDA out of memory. Tried to allocate 72.00 MiB (GPU 0; 3.00 GiB total capacity; 1.84 GiB already allocated; 5.45 MiB free; 2.04 GiB reserved in total by PyTorch)

虽然我没有使用 CUDA 内存，但它仍然保持在同一水平。

我尝试使用torch.cuda.empty_cache()，但它并没有影响问题。当我关闭 jupyter 笔记本时，它减少到 0。所以我很确定它是 pytorch 和 python 的东西。

这是我的代码：

import joblib
import torch
from transformers import AutoTokenizer, AutoModelForSequenceClassification,pipeline
import torch.nn.functional as F
from torch.utils.data import DataLoader
import pandas as pd
import numpy as np
from tqdm import tqdm

tokenizer = AutoTokenizer.from_pretrained("savasy/bert-base-turkish-sentiment-cased")
model …

我想将我的计数器数据和时间导出为 csv 文件，然后我将用它们制作一个 2D 图表。如何将这些记录并写入 csv 文件？