直接按照下图模块测测试调用GPU就好了
这是测试GPU是否能使用。
如果显示了可以使用后续使用操作如下：
将实例化的模型传入GPU中：
直接看一个训练的示例代码：
要在GPU中处理使用都要传入GPU中才能用。
