kyleleey commited on
Commit
650d9ac
1 Parent(s): e14627f

remove tcnn

Browse files
Files changed (2) hide show
  1. Dockerfile +0 -2
  2. video3d/render/mlptexture.py +3 -3
Dockerfile CHANGED
@@ -57,8 +57,6 @@ RUN git clone -b v0.7.4 https://github.com/facebookresearch/pytorch3d.git /home/
57
  WORKDIR /home/user/pytorch3d-0.7.4
58
  RUN python setup.py install --user
59
 
60
- RUN pip install git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch
61
-
62
  # Set the working directory to the user's home directory
63
  WORKDIR $HOME/app
64
 
 
57
  WORKDIR /home/user/pytorch3d-0.7.4
58
  RUN python setup.py install --user
59
 
 
 
60
  # Set the working directory to the user's home directory
61
  WORKDIR $HOME/app
62
 
video3d/render/mlptexture.py CHANGED
@@ -8,7 +8,7 @@
8
  # its affiliates is strictly prohibited.
9
 
10
  import torch
11
- import tinycudann as tcnn
12
  import numpy as np
13
 
14
  #######################################################################################################################################################
@@ -118,5 +118,5 @@ class MLPTexture3D(torch.nn.Module):
118
 
119
  return out.view(*texc.shape[:-1], self.channels) # Remap to [n, h, w, c]
120
 
121
- def cleanup(self):
122
- tcnn.free_temporary_memory()
 
8
  # its affiliates is strictly prohibited.
9
 
10
  import torch
11
+ # import tinycudann as tcnn
12
  import numpy as np
13
 
14
  #######################################################################################################################################################
 
118
 
119
  return out.view(*texc.shape[:-1], self.channels) # Remap to [n, h, w, c]
120
 
121
+ # def cleanup(self):
122
+ # tcnn.free_temporary_memory()