首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >当cuda不可用时,“use_cuda”设置为True。确保CUDA可用或设置use_cuda=False

当cuda不可用时,“use_cuda”设置为True。确保CUDA可用或设置use_cuda=False
EN

Stack Overflow用户
提问于 2020-11-30 14:21:30
回答 3查看 2.3K关注 0票数 0

我正在尝试创建一个Bert模型来对土耳其Lan进行分类。下面是我的代码:

代码语言:javascript
运行
复制
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。我认为每件事都是正确的,但当我运行这个命令时,我得到了以下错误:

代码语言:javascript
运行
复制
'use_cuda' set to True when cuda is unavailable. Make sure CUDA is available or set use_cuda=False.

我该如何准确地解决这个问题呢?

EN

Stack Overflow用户

回答已采纳

发布于 2021-01-06 23:58:33

我遇到了同样的问题。如果有CUDA可用,则将use_cudafp16都设置为True。如果不是,则将两者都设置为False

票数 0
EN
查看全部 3 条回答
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/65068922

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档