force cuda dr

inference.py

@@ -114,7 +114,7 @@ def extract_texmesh(args, model, output_path, device):
     w0 = 1024
     ssaa = 1
     fp16 = True
-    glctx = dr.RasterizeGLContext(output_db=False)
+    glctx = dr.RasterizeCudaContext()
     v_np = vertices.astype(np.float32)
     f_np = triangles.astype(np.int64)
     v = torch.from_numpy(vertices).float().contiguous().to(device)