search

WongKinYiu / yolov7

Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors
GNU General Public License v3.0
13.01k stars 4.11k forks source link

CUDA 11.6 Support #539

Open Elreniel opened 1 year ago

Elreniel commented 1 year ago

I have CUDA 11.6 installed on my server. This CUDA works with PyTorch 1.12 only ( as far as I know). Is there a way to use Yolov7 with CUDA 11.6?

ghost commented 1 year ago

I personally have CUDA 11.3 installed and got Yolov7 working with Pytorch 1.10 Incase you need to reparam your trained model i can tell you that only opset < 12 works. I used opset 11 exporting my model to onnx. So first try it out and otherwise move to a CUDA version which supports Pytorch 1.10

Elreniel commented 1 year ago

I want to train a model from scratch. Also there are other environments in the Server so that downgrading CUDA is not an option right now.

ghost commented 1 year ago

maybe it works with pytorch 1.12 just try it out and make sure to use opset version 11