new

src/models/geometry/render/neural_render.py

@@ -15,9 +15,9 @@ from . import util
 from . import renderutils as ru
 _FG_LUT = None
 
-os.environ["CUDA_HOME"] = "/usr/local/cuda"
-os.environ["PATH"] += os.pathsep + os.path.join(os.environ["CUDA_HOME"], "bin")
-os.environ["LD_LIBRARY_PATH"] = os.environ.get("LD_LIBRARY_PATH", "") + os.pathsep + os.path.join(os.environ["CUDA_HOME"], "lib64")
+os.environ['CUDA_HOME'] = '/usr/local/cuda-12.1'
+os.environ['PATH'] += ':/usr/local/cuda-12.1/bin'
+os.environ['LD_LIBRARY_PATH'] += ':/usr/local/cuda-12.1/lib64'
 def interpolate(attr, rast, attr_idx, rast_db=None):
     return dr.interpolate(
         attr.contiguous(), rast, attr_idx, rast_db=rast_db,

src/models/geometry/rep_3d/flexicubes_geometry.py

@@ -16,7 +16,9 @@ from .dmtet import sdf_reg_loss_batch
 from . import mesh
 import torch.nn.functional as F
 from src.utils import render
-
+os.environ['CUDA_HOME'] = '/usr/local/cuda'
+os.environ['PATH'] += ':/usr/local/cuda/bin'
+os.environ['LD_LIBRARY_PATH'] += ':/usr/local/cuda/lib64'
 def get_center_boundary_index(grid_res, device):
     v = torch.zeros((grid_res + 1, grid_res + 1, grid_res + 1), dtype=torch.bool, device=device)
     v[grid_res // 2 + 1, grid_res // 2 + 1, grid_res // 2 + 1] = True