pytorch 运行torchvision.utils.make_grid(图像)后内核死亡

0dxa2lsx  于 2023-01-26  发布在  其他
关注(0)|答案(2)|浏览(204)

为了简单起见,我将按照PyTorch提供的tutorial创建一个CNN。
然而,当我运行这个特定的代码块时,它显示出相应的行:

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

它以某种方式杀死了内核。这让我很困惑,因为它只是一个简单的函数。
我也浏览了许多与PyTorch有关的SO帖子,但都与Torchvision无关。我发现的最相似的问题是5个月前发布的,没有答案。
如果我需要补充任何信息以使这个问题更清楚,请告诉我。

w6mmgewl

w6mmgewl1#

同样的事情也发生在我身上,在python解释器中运行这些命令时,我得到了一个错误,导致了以下结果:
Error #15: Initializing libiomp5md.dll, but found libiomp5md.dll already initialized
当我添加以下内容时,它起作用了:

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

63lcw9qa2#

我也遇到过类似的问题(在MacOS 12. 6. 2上)。使用Torchvision时内核崩溃。

trainset = torchvision.datasets.CIFAR10(root='./data', train=True,
                                        download=True, transform=transform)   
testset = torchvision.datasets.CIFAR10(root='./data', train=False,
                                       download=True, transform=transform)

classes = ('plane', 'car', 'bird', 'cat',
           'deer', 'dog', 'frog', 'horse', 'ship', 'truck')

batch_size = 64
trainloader = th.utils.data.DataLoader(trainset, batch_size=batch_size,
                                          shuffle=True, num_workers=0)
testloader = th.utils.data.DataLoader(testset, batch_size=batch_size,
                                         shuffle=False, num_workers=0)
dataiter = iter(trainloader)
images, labels = next(dataiter)

我通过重新安装pytorch和torchvision解决了这个问题。在脚本中运行(jupyter笔记本或其他地方):

!conda install --yes pytorch torchvision -c pytorch

相关问题