From 015a8ff99fec8697c854cc51f2e1ccf23b6acdd1 Mon Sep 17 00:00:00 2001 From: flowtone Date: Thu, 14 Mar 2024 19:13:22 +0200 Subject: [PATCH] remove unncessary cuda context --- crmlib/util/renderer.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/crmlib/util/renderer.py b/crmlib/util/renderer.py index 77af091..856ef18 100644 --- a/crmlib/util/renderer.py +++ b/crmlib/util/renderer.py @@ -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)