python无法调用cuda

Question

--use-cuda 无GPU环境，请改为False
如何解决这个问题

Answer 1

你可以在使用 transformers 库加载模型时，将 use_cuda 参数设为 False。例如：

import transformers
model = transformers.BertModel.from_pretrained('bert-base-uncased', use_cuda=False)