Update src/models/geometry/render/neural_render.py

src/models/geometry/render/neural_render.py

@@ -14,7 +14,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")
 def interpolate(attr, rast, attr_idx, rast_db=None):
     return dr.interpolate(
         attr.contiguous(), rast, attr_idx, rast_db=rast_db,