• Windows 10
• RTX 3070
• CUDA 11.1
• cuDNN 8.0.5 (适用于CUDA 11.1)
• Python 3.8.5
• tf-nightly-gpu 2.5.0.dev20201113
• 使用Anaconda环境

我终于设法让tf-nightly-gpu在我的3070上运行了，然而，在训练过程中似乎性能呈指数级下降。开始时每个epoch耗时50秒，后来增加到超过700秒一个epoch。在任务管理器中，看起来3070在性能下降一段时间后几乎没有被使用。我还没有尝试任何解决方法，因为我不知道该做什么。有没有建议？

这是我的完整代码和输出：

...

输出：

...

回答：

tensorflow不支持CUDA 11.1，请改用CUDA 11.0，并将CUDA 11.1目录中的ptxas.exe移动到您的11.0目录中。