Traceback (most recent call last):
  File "/home/star/whaiDir/PFCFuse/train.py", line 9, in <module>
    from net import Restormer_Encoder, Restormer_Decoder, BaseFeatureExtraction, DetailFeatureExtraction
  File "/home/star/whaiDir/PFCFuse/net.py", line 1, in <module>
    import torch
  File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/__init__.py", line 2143, in <module>
    from . import _meta_registrations
  File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/_meta_registrations.py", line 9, in <module>
    from torch._decomp import (
  File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/_decomp/__init__.py", line 245, in <module>
    import torch._decomp.decompositions
  File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/_decomp/decompositions.py", line 12, in <module>
    import torch._prims as prims
  File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/_prims/__init__.py", line 844, in <module>
    ndtri = _make_elementwise_unary_prim(
  File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/_prims/__init__.py", line 472, in _make_elementwise_unary_prim
    return _make_prim(
  File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/_prims/__init__.py", line 320, in _make_prim
    prim_def = torch.library.custom_op(
  File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/_library/custom_ops.py", line 142, in custom_op
    return inner(fn)
  File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/_library/custom_ops.py", line 123, in inner
    result = CustomOpDef(namespace, opname, schema_str, fn)
  File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/_library/custom_ops.py", line 169, in __init__
    self._register_to_dispatcher()
  File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/_library/custom_ops.py", line 455, in _register_to_dispatcher
    lib.define(
  File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/library.py", line 118, in define
    result = self.m.define(schema, alias_analysis, tuple(tags))
KeyboardInterrupt