search

IDEA-Research / detrex

detrex is a research platform for DETR-based object detection, segmentation, pose estimation and other visual recognition tasks.
https://detrex.readthedocs.io/en/latest/
Apache License 2.0
1.95k stars 204 forks source link

help,could you install the detectron2 with cuda 11.3? #246

Closed CasonTsai closed 1 year ago

CasonTsai commented 1 year ago

hello ,i have met a fault below when i use this comand 'python -m pip install -e detectron2' to install detectron2

The detected CUDA version (11.3) mismatches the version that was used to compile PyTorch (10.2). Please make sure to use the same CUDA versions. image

my cuda version is 11.3 which is recommended

CasonTsai commented 1 year ago

pytorch version is 1.12

rentainhe commented 1 year ago

It seems you should update your pytorch version, you should install the version complied with cuda11.3 in pytorch official website

CasonTsai commented 1 year ago

i have fixed it ,the reason is that when install torch with pip should down the torch which is pre-builded with the same cuda version