-
Notifications
You must be signed in to change notification settings - Fork 54
Description
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
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