如何使用 GPU 而不是 CPU 训练我的 ChatterBot model

Question

我正在使用 chatterbot 库来开发聊天机器人。 我想使用我的 GPU 而不是我的 CPU 来训练聊天机器人

这是我的代码：

from chatterbot import ChatBot
from chatterbot.trainers import ChatterBotCorpusTrainer


bot = ChatBot(
    #Bot name
    "Elon"
)

trainer = ChatterBotCorpusTrainer(bot)

trainer.train(
    'chatterbot.corpus.english'
)

while True:

    user_input = input('You: ')
    bot_response = bot.get_response(user_input)
    print(bot_response)

Answer 1

查看chatterbot的文档，没有提到这一点。 但也许您可以查看它所构建的框架并尝试更改源代码以允许 GPU 训练。