Pytorch iter问题

最新推荐文章于 2022-12-01 22:37:59 发布

原创最新推荐文章于 2022-12-01 22:37:59 发布 · 3k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #iter

PyTorch 专栏收录该内容

16 篇文章

订阅专栏

本文详细介绍了在使用PyTorch进行深度学习项目时遇到的RuntimeError: randperm is only implemented for CPU错误，并提供了解决方案。通过修改torch.utils.data.sampler.py中的RandomSampler类的__iter__函数，将设备指定为CPU，从而避免了在GPU上运行randperm操作导致的错误。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Pytorch:RuntimeError: randperm is only implemented for CPU

在这里插入图片描述

/anaconda3/lib/python3.6/site-packages/torch/utils/data/sampler.py文件中的__iter__函数改为如下形式:

class RandomSampler(Sampler):
    r"""Samples elements randomly, without replacement.

    Arguments:
        data_source (Dataset): dataset to sample from
    """

    def __init__(self, data_source):
        self.data_source = data_source
        #print('torch.device:',torch.device)
    def __iter__(self):
        cpu = torch.device('cpu',0)
        return iter(torch.randperm(len(self.data_source),device=cpu).tolist())

    def __len__(self):
        return len(self.data_source)