关于#pytorch PYg库#的问题，如何解决？(语言-python)

Question

pytorch1.3.0 cuda10.1 安装了对应版本的Pyg(torch-geometric)相关依赖
测试torch-geometric代码：

import torch
from torch import Tensor
from torch_geometric.nn import GCNConv
from torch_geometric.datasets import Planetoid

dataset = Planetoid(root='.', name='Cora')

class GCN(torch.nn.Module):
    def __init__(self, in_channels, hidden_channels, out_channels):
        super().__init__()
        self.conv1 = GCNConv(in_channels, hidden_channels)
        self.conv2 = GCNConv(hidden_channels, out_channels)

    def forward(self, x: Tensor, edge_index: Tensor) -> Tensor:
        # x: Node feature matrix of shape [num_nodes, in_channels]
        # edge_index: Graph connectivity matrix of shape [2, num_edges]
        x = self.conv1(x, edge_index).relu()
        x = self.conv2(x, edge_index)
        return x

model = GCN(dataset.num_features, 16, dataset.num_classes)
print(model)

发生报错

Traceback (most recent call last):
  File "/code/Source2020/test/test_torch_geometric.py", line 3, in <module>
    from torch_geometric.nn import GCNConv
  File "/envs/torch1.3.0_torch_geometric1.3.2_py37/lib/python3.7/site-packages/torch_geometric/__init__.py", line 2, in <module>
    import torch_geometric.nn
  File "/envs/torch1.3.0_torch_geometric1.3.2_py37/lib/python3.7/site-packages/torch_geometric/nn/__init__.py", line 2, in <module>
    from .data_parallel import DataParallel
  File "/envs/torch1.3.0_torch_geometric1.3.2_py37/lib/python3.7/site-packages/torch_geometric/nn/data_parallel.py", line 5, in <module>
    from torch_geometric.data import Batch
  File "/envs/torch1.3.0_torch_geometric1.3.2_py37/lib/python3.7/site-packages/torch_geometric/data/__init__.py", line 1, in <module>
    from .data import Data
  File "/envs/torch1.3.0_torch_geometric1.3.2_py37/lib/python3.7/site-packages/torch_geometric/data/data.py", line 7, in <module>
    from torch_sparse import coalesce
  File "/envs/torch1.3.0_torch_geometric1.3.2_py37/lib/python3.7/site-packages/torch_sparse/__init__.py", line 2, in <module>
    from .coalesce import coalesce
  File "/envs/torch1.3.0_torch_geometric1.3.2_py37/lib/python3.7/site-packages/torch_sparse/coalesce.py", line 2, in <module>
    import torch_scatter
  File "/envs/torch1.3.0_torch_geometric1.3.2_py37/lib/python3.7/site-packages/torch_scatter/__init__.py", line 3, in <module>
    from .mul import scatter_mul
  File "/envs/torch1.3.0_torch_geometric1.3.2_py37/lib/python3.7/site-packages/torch_scatter/mul.py", line 3, in <module>
    from torch_scatter.utils.ext import get_func
  File "/envs/torch1.3.0_torch_geometric1.3.2_py37/lib/python3.7/site-packages/torch_scatter/utils/ext.py", line 5, in <module>
    import torch_scatter.scatter_cuda
ImportError: /envs/torch1.3.0_torch_geometric1.3.2_py37/lib/python3.7/site-packages/torch_scatter/scatter_cuda.cpython-37m-x86_64-linux-gnu.so: undefined symbol: cudaSetupArgument

附带conda环境中已有的包

不知道什么地方有问题，求指点！