ModuleNotFoundError: No module named 'dgl.nn.tensorflow'
图神经网络包dgl支持pytorch、tensorflow、mxnet三个深度学习框架,但如果在tensorflow框架下调用dgl.nn.tensorflow
经常会报错:
> ModuleNotFoundError: No module named 'dgl.nn.tensorflow'
这是因为DGL默认的backend是pytorch,如果在tensorflow或者mxnet框架下使用则需要把backend修改为对应的名称。windows下修改方式:
- 文件资源管理器地址栏输入
%HOMEPATH%
,然后回车。 - 在目录中打开
.dgl
目录,打开config.json
- 把对应的内容修改为
{"backend": "tensorflow"}
- 保存文件,问题解决。