Update src/models/geometry/render/renderutils/ops.py

src/models/geometry/render/renderutils/ops.py

@@ -9,6 +9,9 @@
 
 import numpy as np
 import os
+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'
 import sys
 import torch
 import torch.utils.cpp_extension