gpt4 book ai didi

python - 有没有办法在没有 GPU 和 TensorFlow 的情况下运行 GPT2

转载 作者:太空宇宙 更新时间:2023-11-03 19:55:07 27 4
gpt4 key购买 nike

GPT2是一个优秀的NLP OpenAI项目。开发人员要求指出我们需要使用 tensorflow 和 GPU。我只想使用(而不是训练)现有的训练参数。有没有办法使用GPT2而不需要昂贵的GPU硬件并且不需要安装Tensor Flow?

最佳答案

我会做什么:

  • sess.run([var for var in tf.trainable_variables]) 获取经过训练的参数并将其保存为 numpy 数组。
  • 重建仅 numpy 版本的 gpt2。

但是:

  • 重建模型需要大量工作
  • 您需要至少安装一次tensorflow才能获取经过训练的变量。

至于gpu,实际上并不需要它,模型将使用cpu版本的tensorflow运行(特别需要gpu来加快训练时间)。

关于python - 有没有办法在没有 GPU 和 TensorFlow 的情况下运行 GPT2,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/59588423/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com