没有tpu，使用GPU训练模型

[lzf00]

我跑通了模型，但是默认是用cpu运行的，我想使用GPU运行，但是不知道在那个文件改什么参数？能指导一下吗，感谢！

[juntaoy]

用gpu还是cpu取决于你安装的tensor flow的版本还有gpu是不是有cuda和cudnn不用在我的代码里做设置的

…

On 3 Sep 2022, at 10:36, lzf1024 ***@***.***> wrote:  我跑通了模型，但是默认是用cpu运行的，我想使用GPU运行，但是不知道在那个文件改什么参数？能指导一下吗，感谢！ — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you are subscribed to this thread.