为什么main的新版本，我开启训练后只能使用CPU，没法使用GPU，训练速度低到 0.08setp/s

[syy7854]

为什么main的新版本，我开启训练后只能使用CPU，没法使用GPU，训练速度低到 0.08setp/s

{| Epoch: 52/3334 (3/3) | Loss: 1.005 | 0.081 steps/s | Step: 0k | } {| Epoch: 53/3334 (3/3) | Loss: 0.9955 | 0.081 steps/s | Step: 0k | } {| Epoch: 54/3334 (3/3) | Loss: 0.9905 | 0.081 steps/s | Step: 0k | } {| Epoch: 55/3334 (3/3) | Loss: 0.9797 | 0.081 steps/s | Step: 0k | }

任务管理器中，GPU显示为3%，CPU为60%

是还需要修改什么参数吗

[babysor]

pytorch gpu版没安装对，查看下

[HaSaKiYasuooo]

怎样才算安装正确，如何匹配！

[L1590]

import torch torch.cuda.is_available() 若返回为True，则使用的是GPU版本的torch，若为False，则为CPU版本