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

Question

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

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

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

use_cuda= False

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

Answer 3

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