为什么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%
是还需要修改什么参数吗
3条答案
按热度按时间cuxqih211#
pytorch gpu版没安装对，查看下
pxq42qpu2#
怎样才算安装正确，如何匹配！
a5g8bdjr3#
import torch torch.cuda.is_available()若返回为True，则使用的是GPU版本的torch，若为False，则为CPU版本