installation package #17
Description
ImportError Traceback (most recent call last)
/tmp/ipykernel_3103551/2851541913.py in
----> 1 from parallel import DataParallel
2 from eval import eval_model
~/my_code/git/RGM/parallel/init.py in
----> 1 from .data_parallel import *
2
3 all = ['DataParallel']
~/my_code/git/RGM/parallel/data_parallel.py in
5 import torch
6 import torch.nn as nn
----> 7 from .scatter_gather import scatter_kwargs, gather
8
9
~/my_code/git/RGM/parallel/scatter_gather.py in
1 import torch
2 import torch.nn.parallel.scatter_gather as torch_
----> 3 from sparse_torch import CSRMatrix3d, CSCMatrix3d, concatenate
4
5
~/my_code/git/RGM/sparse_torch/init.py in
3 """
4
----> 5 from .csx_matrix import CSRMatrix3d, CSCMatrix3d, dot, concatenate
~/my_code/git/RGM/sparse_torch/csx_matrix.py in
6 from torch.utils.cpp_extension import load
7
----> 8 sparse_dot = load(name='sparse_dot',
9 sources=['extension/sparse_dot/sparse_dot.cpp',
10 'extension/sparse_dot/csr_dot_csc_cuda.cu',
~/.local/lib/python3.9/site-packages/torch/utils/cpp_extension.py in load(name, sources, extra_cflags, extra_cuda_cflags, extra_ldflags, extra_include_paths, build_directory, verbose, with_cuda, is_python_module, is_standalone, keep_intermediates)
1200 verbose=True)
1201 '''
-> 1202 return _jit_compile(
1203 name,
1204 [sources] if isinstance(sources, str) else sources,
~/.local/lib/python3.9/site-packages/torch/utils/cpp_extension.py in _jit_compile(name, sources, extra_cflags, extra_cuda_cflags, extra_ldflags, extra_include_paths, build_directory, verbose, with_cuda, is_python_module, is_standalone, keep_intermediates)
1448 return _get_exec_path(name, build_directory)
1449
-> 1450 return _import_module_from_library(name, build_directory, is_python_module)
1451
1452
~/.local/lib/python3.9/site-packages/torch/utils/cpp_extension.py in _import_module_from_library(module_name, path, is_python_module)
1842 spec = importlib.util.spec_from_file_location(module_name, filepath)
1843 assert spec is not None
-> 1844 module = importlib.util.module_from_spec(spec)
1845 assert isinstance(spec.loader, importlib.abc.Loader)
1846 spec.loader.exec_module(module)
ImportError: /home/jupyter-lijianguo/.cache/torch_extensions/py39_cu116/sparse_dot/sparse_dot.so: cannot open shared object file: No such file or directory