nksr/_C.so: undefined symbol: _ZN2at4_ops11linalg_norm4callERKNS_6TensorERKN3c108optionalINS5_6ScalarEEENS5_16OptionalArrayRefIlEEbNS6_INS5_10ScalarTypeEEE

I execute the running command: (xcube) $ python inference/sample_shapenet.py none --category car --total_len 20 --batch_len 4 --ema --use_ddim --ddim_step 100 --extract_mesh

But the terminal prompts:

/pfs/mt-1oY5F7/lihong/miniconda3/envs/xcube/lib/python3.10/site-packages/pytorch_lightning/utilities/distributed.py:258: LightningDeprecationWarning: `pytorch_lightning.utilities.distributed.rank_zero_only` has been deprecated in v1.8.1 and will be removed in v2.0.0. You can import it from `pytorch_lightning.utilities` instead.
  rank_zero_deprecation(
2024-07-13 17:30:49.739 | INFO     | xcube.modules.autoencoding.sunet:__init__:240 - latent dim: 16
2024-07-13 17:30:51.765 | INFO     | xcube.models.diffusion:__init__:131 - num_input_channels: 16, out_channels: 16,                     num_classes: None, context_dim: None, concat_dim: None                     conditioning_key: none
2024-07-13 17:31:08.413 | INFO     | xcube.models.diffusion:__init__:247 - Keeping EMAs of 416.
2024-07-13 17:31:08.414 | INFO     | xcube.models.diffusion:__init__:255 - Manully setting scale_factor to 1.0
2024-07-13 17:31:31.158 | INFO     | xcube.modules.autoencoding.sunet:__init__:240 - latent dim: 8
2024-07-13 17:31:31.343 | INFO     | xcube.models.diffusion:__init__:131 - num_input_channels: 14, out_channels: 8,                     num_classes: None, context_dim: None, concat_dim: None                     conditioning_key: concat
2024-07-13 17:31:37.561 | INFO     | xcube.models.diffusion:__init__:247 - Keeping EMAs of 386.
2024-07-13 17:31:37.567 | INFO     | xcube.models.diffusion:__init__:255 - Manully setting scale_factor to 1.0
Traceback (most recent call last):
  File "/pfs/mt-1oY5F7/lihong/nips/XCube/inference/sample_shapenet.py", line 90, in <module>
    import nksr
  File "/pfs/mt-1oY5F7/lihong/miniconda3/envs/xcube/lib/python3.10/site-packages/nksr/__init__.py", line 18, in <module>
    from nksr.nn.unet import SparseStructureNet
  File "/pfs/mt-1oY5F7/lihong/miniconda3/envs/xcube/lib/python3.10/site-packages/nksr/nn/__init__.py", line 10, in <module>
    from .modules import Conv3d, GroupNorm, Activation, GroupNorm, MaxPooling, Upsampling, SparseZeroPadding
  File "/pfs/mt-1oY5F7/lihong/miniconda3/envs/xcube/lib/python3.10/site-packages/nksr/nn/modules.py", line 14, in <module>
    from nksr.svh import SparseFeatureHierarchy, KernelMap, VoxelStatus
  File "/pfs/mt-1oY5F7/lihong/miniconda3/envs/xcube/lib/python3.10/site-packages/nksr/svh.py", line 14, in <module>
    from nksr.ext import _CpuIndexGrid, _CudaIndexGrid
  File "/pfs/mt-1oY5F7/lihong/miniconda3/envs/xcube/lib/python3.10/site-packages/nksr/ext.py", line 9, in <module>
    from nksr import _C
ImportError: /pfs/mt-1oY5F7/lihong/miniconda3/envs/xcube/lib/python3.10/site-packages/nksr/_C.so: undefined symbol: _ZN2at4_ops11linalg_norm4callERKNS_6TensorERKN3c108optionalINS5_6ScalarEEENS5_16OptionalArrayRefIlEEbNS6_INS5_10ScalarTypeEEE

nv-tlabs / XCube

nksr/_C.so: undefined symbol: _ZN2at4_ops11linalg_norm4callERKNS_6TensorERKN3c108optionalINS5_6ScalarEEENS5_16OptionalArrayRefIlEEbNS6_INS5_10ScalarTypeEEE #10