detectron2 icon indicating copy to clipboard operation
detectron2 copied to clipboard

GPU is not used when running demo.py

Open MasterMiao919 opened this issue 3 years ago • 0 comments

Run command

python demo.py --config-file ../configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml --video-input test.mp4 --opts MODEL.WEIGHTS detectron2://COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl

Expected behavior

When running demo.py, the GPU is not used, but the CPU is used

image

Environment

sys.platform win32 Python 3.7.13 (default, Mar 28 2022, 08:03:21) [MSC v.1916 64 bit (AMD64)] numpy 1.21.5 detectron2 0.6 @L:\Networks\detectron2\detectron2 Compiler MSVC 192930145 CUDA compiler CUDA 11.6 detectron2 arch flags L:\Networks\detectron2\detectron2_C.pyd; cannot find cuobjdump DETECTRON2_ENV_MODULE PyTorch 1.11.0+cu113 @P:\Anaconda3\envs\pytorch\lib\site-packages\torch PyTorch debug build False GPU available Yes GPU 0 NVIDIA GeForce RTX 3070 (arch=8.6) Driver version 511.65 CUDA_HOME C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6 Pillow 9.1.0 torchvision 0.12.0+cu113 @P:\Anaconda3\envs\pytorch\lib\site-packages\torchvision torchvision arch flags P:\Anaconda3\envs\pytorch\lib\site-packages\torchvision_C.pyd; cannot find cuobjdump fvcore 0.1.5 iopath 0.1.9 cv2 4.5.5

MasterMiao919 avatar Jul 27 '22 01:07 MasterMiao919