RuntimeError: CUDA error: out of memory in the demo.py

[Hikeylin]

Config '../configs/SSIS/MS_R_101_BiFPN_with_offset_class_demo.yaml' has no VERSION. Assuming it to be compatible with latest v2.
Traceback (most recent call last):
File "/home/w/SSIS/demo/demo.py", line 79, in
demo = VisualizationDemo(cfg)
File "/home/w/SSIS/demo/predictor.py", line 44, in init
self.predictor = DefaultPredictor(cfg)
File "/home/w/anaconda3/envs/ssis/lib/python3.9/site-packages/detectron2/engine/defaults.py", line 282, in init
self.model = build_model(self.cfg)
File "/home/w/anaconda3/envs/ssis/lib/python3.9/site-packages/detectron2/modeling/meta_arch/build.py", line 22, in build_model
model = META_ARCH_REGISTRY.get(meta_arch)(cfg)
File "/home/w/SSIS/adet/modeling/ssis/condinst.py", line 73, in init
pixel_mean = torch.Tensor(cfg.MODEL.PIXEL_MEAN).to(self.device).view(3, 1, 1)
RuntimeError: CUDA error: out of memory
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.