运行 torchvision.utils.make_grid(images) 后死内核

Question

为了简单起见，我按照 PyTorch 提供的本教程来创建 CNN。

然而，当我使用相应的行运行这个特定的代码块时：

# show images
imshow(torchvision.utils.make_grid(images))

它以某种方式杀死了内核。这让我很困惑，因为它只是一个简单的函数。

我还浏览了许多与 PyTorch 相关的 SO 帖子，但所有这些帖子都与 torchvision 无关。我发现的最相似的问题是 5 个月前发布的，但没有答案。

如果我需要添加任何信息以使这个问题更清楚，请告诉我。

Answer 1

这样的事情我也经历过。在 python 解释器中运行这些命令时，我收到一个错误，导致我出现以下情况：

错误#15：正在初始化libiomp5md.dll，但发现libiomp5md.dll已经初始化

当我添加以下内容时，它起作用了：

import os
os.environ['KMP_DUPLICATE_LIB_OK']='True'