'use_cuda' 当 cuda 不可用时设置为 True。确保 CUDA 可用或设置 use_cuda=False

Question

我正在尝试创建一个用于对土耳其语 Lan 进行分类的 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.

我该如何正确解决这个问题？

Answer 1

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

Answer 2

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

Answer 3

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

如果你的电脑没有GPU，这个错误就会抛给你。不要忘记包含此参数

use_cuda= False

这不会影响您的结果，只是比平时多花几秒钟来处理。

'use_cuda' 当 cuda 不可用时设置为 True。确保 CUDA 可用或设置 use_cuda=False

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

python

torch

simpletransformers