Search code examples
pythontorchsimpletransformers

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

I am trying to create a Bert model for classifying Turkish Lan. here is my code:

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
)

I am using Anaconda and Spyder. I think every thing is correct but when I run this I got the following error:

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

how can I fix this exactly?

Solution

  • I ran into the same problem. If you have CUDA available, then set both use_cuda and fp16 to True. If not, then set both to False.