diff --git a/modelscope/models/cv/nerf_recon_4k/network/utils.py b/modelscope/models/cv/nerf_recon_4k/network/utils.py
index aa8c3d66..6983d36a 100644
--- a/modelscope/models/cv/nerf_recon_4k/network/utils.py
+++ b/modelscope/models/cv/nerf_recon_4k/network/utils.py
@@ -5,7 +5,6 @@ from collections import defaultdict
 
 import mcubes
 import numpy as np
-import tinycudann as tcnn
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
@@ -166,6 +165,8 @@ def normalize(dat, inp_scale, tgt_scale):
 
 
 def cleanup():
+    import tinycudann as tcnn
+
     gc.collect()
     torch.cuda.empty_cache()
     tcnn.free_temporary_memory()
diff --git a/modelscope/models/cv/nerf_recon_acc/network/nerf.py b/modelscope/models/cv/nerf_recon_acc/network/nerf.py
index 972d2d85..64ffae14 100644
--- a/modelscope/models/cv/nerf_recon_acc/network/nerf.py
+++ b/modelscope/models/cv/nerf_recon_acc/network/nerf.py
@@ -1,7 +1,6 @@
 # The implementation is modified from nerfacc, made publicly available under the MIT License
 # at https://github.com/KAIR-BAIR/nerfacc/blob/master/examples/radiance_fields/ngp.py
 import numpy as np
-import tinycudann as tcnn
 import torch
 import torch.nn as nn
 from nerfacc import ContractionType, OccupancyGrid, ray_marching, rendering
@@ -113,6 +112,8 @@ trunc_exp = _TruncExp.apply
 class VolumeDensity(nn.Module):
 
     def __init__(self, config):
+        import tinycudann as tcnn
+
         super().__init__()
         self.config = config
         self.radius = self.config.radius
diff --git a/modelscope/models/cv/nerf_recon_acc/network/utils.py b/modelscope/models/cv/nerf_recon_acc/network/utils.py
index aa8c3d66..6983d36a 100644
--- a/modelscope/models/cv/nerf_recon_acc/network/utils.py
+++ b/modelscope/models/cv/nerf_recon_acc/network/utils.py
@@ -5,7 +5,6 @@ from collections import defaultdict
 
 import mcubes
 import numpy as np
-import tinycudann as tcnn
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
@@ -166,6 +165,8 @@ def normalize(dat, inp_scale, tgt_scale):
 
 
 def cleanup():
+    import tinycudann as tcnn
+
     gc.collect()
     torch.cuda.empty_cache()
     tcnn.free_temporary_memory()