Question

在dgl（DeepMind的Deep Graph Library）中导入pytorch时出现错误，具体是：

ImportError：/home/user/anaconda3/envs/my_env/lib/python3.7/site-packages/torch/lib/libtorch.so.1：未定义的符号：nvrtcGetProgramLogSize

我试图重新安装pytorch（使用/etc/config/log_level卸载重新安装）。我也在Google中搜索，发现了这个https://github.com/pytorch/pytorch/issues/14973。在那里，他们解决了在conda un/install和libnvrtc.so中进行链接的问题，但是我不知道那是什么意思。有人知道吗？

这是基本代码：

libcuda.so

这是错误：

import dgl
from parseador import train_df

g = dgl.DGLGraph()
g.add_nodes(5)
g.add_edges([0, 0, 0, 0], [1, 2, 3, 4])
g.ndata['h'] = th.randn(5, 3)
g.edata['h'] = th.randn(4, 4)

如何解决此错误？前一段时间，我在Windows 10中正确运行了此代码，而现在我正在运行Ubuntu 18.04。

Answer 1

我也遇到了这个问题，但是我实际上想使用GPU，因此对我来说，安装pytorch-cpu并不是一种选择。

相反，从pytorch频道安装pytorch软件包（而不是defaults）为我解决了这个问题：conda install pytorch --channel pytorch

Answer 2

我刚刚使用conda install pytorch-cpu

修复了相同的异常

Answer 3

您应在GPU机器上运行代码，并确保pytorch是GPU

如何在DGL中修复“导入错误：/ home / ... ... / lib / libtorch.so.1：未定义的符号：nvrtcGetProgramLogSize”？

3 个答案: