search

elyha7 / yoloface

Yolov5 Face Detection
GNU General Public License v3.0
119 stars 37 forks source link

NVIDIA GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation. #14

Open Stephenfang51 opened 11 months ago

Stephenfang51 commented 11 months ago

Can't get attribute 'SPPF' on <module 'models.common' from when model = torch.load('yolov5m.pt', map_location='cpu')['model']

winston1214 commented 8 months ago

I think you must install torch stable version. For example, if you use torch==1.7.0

pip install torch==1.7.0+cu110 torchvision==0.8.0+cu110 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html