Test MIG support

cpn.py

@@ -58,8 +58,8 @@ class CpnInterface:
         with torch.no_grad():
             out = cd.asnumpy(self.model(x, crop_size=self.tile_size,
                                         stride=max(64, self.tile_size - self.overlap)))
-            # if torch.cuda.device_count():
-            #     print(cd.GpuStats())
+            if torch.cuda.device_count():
+                print(cd.GpuStats())
 
         contours, = out['contours']
         boxes, = out['boxes']