RuntimeError: module must have its parameters and buffers on device cuda:0 (device_ids[0]) but found

最新推荐文章于 2024-04-24 23:47:47 发布

Andy_2259

最新推荐文章于 2024-04-24 23:47:47 发布

阅读量1.2k

点赞数 1

分类专栏： Python 文章标签：深度学习 python 人工智能

本文链接：https://blog.youkuaiyun.com/m0_37690430/article/details/127323928

版权

Python 专栏收录该内容

52 篇文章

订阅专栏

RuntimeError: module must have its parameters and buffers on device cuda:0 (device_ids[0]) but found one of them on device: cpu

nn.Sequential(
                nn.Conv2d(3, 512, kernel_size=1, stride=1, padding=0, bias=False),
                nn.BatchNorm2d(512), nn.ReLU(inplace=True),
                nn.AdaptiveAvgPool2d((1, 1)),
                nn.Flatten(),
                )

在模型最后加.cuda()
改为：

nn.Sequential(
                nn.Conv2d(3, 512, kernel_size=1, stride=1, padding=0, bias=False),
                nn.BatchNorm2d(512), nn.ReLU(inplace=True),
                nn.AdaptiveAvgPool2d((1, 1)),
                nn.Flatten(),
                ).cuda()