Tensorflow未完全利用GPU在GPT-2程序中

2024-09-25 08:32:51 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在运行大型模型(774M)的GPT-2代码。它是用来生成文本样本,通过交互式的_样本.py,链接:here

因此,我给出了一个输入文件,其中包含自动选择生成输出的提示。此输出也会自动复制到文件中。简而言之,我不是在训练它,而是在使用模型生成文本。 而且,我使用的是一个GPU。你知道吗

我在这方面面临的问题是,代码没有充分利用GPU。你知道吗

通过使用nvidiasmi命令,我可以看到下面的图像

https://imgur.com/CqANNdB


Tags: 文件代码pyhttps模型图像文本命令