0

我正在尝试创建一个用于对土耳其语进行分类的 Bert 模型。这是我的代码：

import pandas as pd
import torch
df = pd.read_excel (r'preparedDataNoId.xlsx')
df = df.sample(frac = 1)

from sklearn.model_selection import train_test_split

train_df, test_df = train_test_split(df, test_size=0.10)

print('train shape: ',train_df.shape)
print('test shape: ',test_df.shape)
from simpletransformers.classification import ClassificationModel

# define hyperparameter
train_args ={"reprocess_input_data": True,
             "fp16":False,
             "num_train_epochs": 4}

# Create a ClassificationModel
model = ClassificationModel(
    "bert", "dbmdz/bert-base-turkish-cased",
    num_labels=4,
    args=train_args
)

我正在使用 Anaconda 和 Spyder。我认为每件事都是正确的，但是当我运行它时，出现以下错误：

'use_cuda' set to True when cuda is unavailable. Make sure CUDA is available or set use_cuda=False.

我该如何解决这个问题？

4

3 回答 3

0

我遇到了同样的问题。如果您有可用的 CUDA，则将use_cuda和都设置fp16为True. 如果不是，则将两者都设置为False。

于 2021-01-06T15:58:33.093 回答
0

CUDA 是 Nvidia 开发的一种并行计算平台和编程模型，用于在自己的 GPU 上进行通用计算。

如果您的计算机没有 GPU，则会向您抛出此错误。不要忘记包含此参数

use_cuda= False

这不会影响您的结果，只需比平时多几秒钟即可处理。

于 2021-08-01T07:05:08.457 回答
-1

如果您的 GPU 在您的计算机上不可用。确保检查 CUDA 或尝试use_cuda=False使用模型的 args。由于您的计算机上不存在 CUDA，因此将引发此错误。

于 2021-06-09T03:13:05.873 回答