remove unncessary cuda context

2024-03-14 19:13:22 +02:00
parent 2b7d2b3e96
commit 015a8ff99f
1 changed files with 0 additions and 4 deletions
@@ -14,10 +14,6 @@ class Renderer(nn.Module):
        self.scale = scale
        self.geo_type = geo_type

-        if torch.cuda.is_available():
-            import nvdiffrast.torch as dr
-            self.glctx = dr.RasterizeCudaContext()
-
        if self.geo_type == "flex":
            device = get_torch_device() if torch.cuda.is_available() else "cpu"
            self.flexicubes = FlexiCubesGeometry(grid_res = self.tet_grid_size, device=device)