导入 Torch 库后,在绘制图形时内核不断死亡

New*_*eAF 7 python matplotlib jupyter-notebook pytorch

我正在尝试运行以下代码:

import matplotlib.pyplot as plt
%matplotlib inline
import torch

x = y = torch.tensor([1,2,3]).numpy()
plt.plot(x,y);
Run Code Online (Sandbox Code Playgroud)

我不断收到消息:The kernel appears to have died. It will restart automatically.工具栏上有一个重新启动和一个红色的“死内核”标签。

但奇怪的是,如果我先导入matplotlib.pyplot并绘制一些随机图,上面的代码绘制得很好。换句话说,以下代码工作正常。

import matplotlib.pyplot as plt
%matplotlib inline

plt.subplots(figsize=(0.01,0.01))
plt.gca().set_visible(False);

import torch
x = torch.tensor([1,2,3]).numpy()

plt.plot(x,x);
Run Code Online (Sandbox Code Playgroud)

这里发生了什么?如果重要的话,我在 Python 3.7.7 上使用 numpy 1.18.5、pytorch 1.6.0、matplotlib 3.2.2。谢谢你。

小智 4

import os
os.environ["KMP_DUPLICATE_LIB_OK"]="TRUE"
Run Code Online (Sandbox Code Playgroud)

首先运行它,然后它将解决您的问题。虽然我猜测,这是一个临时解决方案,您可以参考此链接: https: //www.programmersought.com/article/53286415201/