解决pytorch莫名报错Segmentation fault (core dumped)

问题简述

事情是这样的，今天本来想用pytorch提取一下图片特征：

def exctract_resnet50(pic_dir):
    model = resnet50(weights=ResNet50_Weights.DEFAULT)
    model.eval()
    model = torch.nn.Sequential(*list(model.children())[:-1]).cuda(0)

    to