LTT commited on
Commit
8185e57
1 Parent(s): c99d3f5

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

Browse files
src/models/geometry/render/neural_render.py CHANGED
@@ -14,7 +14,9 @@ from . import util
14
  from . import renderutils as ru
15
  _FG_LUT = None
16
 
17
-
 
 
18
  def interpolate(attr, rast, attr_idx, rast_db=None):
19
  return dr.interpolate(
20
  attr.contiguous(), rast, attr_idx, rast_db=rast_db,
 
14
  from . import renderutils as ru
15
  _FG_LUT = None
16
 
17
+ os.environ["CUDA_HOME"] = "/usr/local/cuda"
18
+ os.environ["PATH"] += os.pathsep + os.path.join(os.environ["CUDA_HOME"], "bin")
19
+ os.environ["LD_LIBRARY_PATH"] = os.environ.get("LD_LIBRARY_PATH", "") + os.pathsep + os.path.join(os.environ["CUDA_HOME"], "lib64")
20
  def interpolate(attr, rast, attr_idx, rast_db=None):
21
  return dr.interpolate(
22
  attr.contiguous(), rast, attr_idx, rast_db=rast_db,