python - QuestionAnsweringModel 不适用于 GPU

Question

我正在使用 SimpleTransformers 的 QuestionAnsweringModel。当我运行我的代码并在 Windows 任务管理器中查看我的进程时，python 根本没有使用 GPU。我已经包含了一个代码片段来重新创建问题。非常感谢任何帮助。

import torch
from simpletransformers.question_answering 
import QuestionAnsweringModel,QuestionAnsweringArgs

model_type=“bert”
model_name= “bert-base-cased”
model_args = QuestionAnsweringArgs()
train_args = {
'n_best_size':1 ,
‘overwrite_output_dir’: True,
‘show_running_loss’:True,
‘n_gpu’: 3

}
model = QuestionAnsweringModel(model_type,model_name, args=train_args, use_cuda=True)

我之前也在这里查看过关于同一问题的主题。建议更新 pytorch。但我也已经这样做了。

更新：尝试使用下面的代码手动将设备设置为 CUDA，但到目前为止没有运气。

model.to(torch.device(“cuda:0” if torch.cuda.is_available() else “cpu”))

谢谢！