关于gpu调用问题

[lvjinqiao]

大神的代码写的非常精巧，因为我是个小白，我看了您的代码不太理解是怎么调用的gpu，我感觉tensor还是跑在cpu上饿。。。

[tensorfly-gpu]

gpu是装了gpu版本的框架之后，应该是自动调用的gpu的。你在aistudio上可以跑一下我的项目，可以看到显存的占用。

[lvjinqiao]

我打算在自己的pytroch环境下跑，所以可能还是得加cuda调用才行，我先自己改改试试> gpu是装了gpu版本的框架之后，应该是自动调用的gpu的。你在aistudio上可以跑一下我的项目，可以看到显存的占用。