我正在尝试创建一个Bert模型来对土耳其Lan进行分类。下面是我的代码:
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.
我该如何准确地解决这个问题呢?
发布于 2021-01-06 23:58:33
我遇到了同样的问题。如果有CUDA可用,则将use_cuda
和fp16
都设置为True
。如果不是,则将两者都设置为False
。
发布于 2021-08-01 07:05:08
CUDA是由Nvidia开发的并行计算平台和编程模型,用于在其自己的GPU上进行通用计算。
如果您的计算机没有GPU,则会抛出此错误。别忘了包括这个参数
use_cuda= False
这不会影响您的结果,只需比平时多花几秒钟处理即可。
发布于 2021-06-09 11:13:06
如果您的GPU在您的计算机上不可用。确保选中CUDA或在模型的args中尝试use_cuda=False
。由于CUDA在您的计算机上不存在,因此将抛出此错误。
https://stackoverflow.com/questions/65068922
复制相似问题