search

conansherry / detectron2

detectron2 windows build
Apache License 2.0
223 stars 54 forks source link

GPU is not used when running demo.py #33

Open MasterMiao919 opened 2 years ago

MasterMiao919 commented 2 years ago

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