导入torchvision.transforms时分段故障

Question

在导入过程中，我在python中得到了一个片段错误。此代码：

import os
import matplotlib.pyplot as plt
import numpy as np
import torch
from torch import nn
from torch import optim
import torch.nn.functional as F
print("was there")
from torchvision import transforms
print("didn't get there")
from torchvision import datasets
from torchvision import models

返回以下内容：

$ python3 -u classifier.py 
was there
Erreur de segmentation (core dumped)

因此，torchvision.transforms似乎应该对此负责。我试过切换线路，torchvision.models也失败了。

我也尝试过自己导入torchvision.transforms，没有问题。是什么导致了这一切？

编辑：

我正在使用Ubuntu20.04.4并通过pip安装torchvision。

Answer 1

因此，我将torchvision.transforms导入移至matplotlib.pyplot导入的上方，而torchvision.transforms和torchvision.models不知何故不再导致分段错误。torchvision.transforms在matplotlib.pyplot后仍引起节段性损伤。

最后的代码如下所示：

import os
from torchvision import transforms
import matplotlib.pyplot as plt
import numpy as np
import torch
from torch import nn
from torch import optim
import torch.nn.functional as F
from torchvision import datasets
from torchvision import models

至少我的代码起作用了，但我觉得一定有一个根本的问题，这并不是.