一、大家如果使用过不同版本的yolo都会遇到一个问题,那就是在pytorch官网上有各种各样的torch和cuda的版本,到底是去选那个好。
下面的链接是pytorch官网链接。
二、我给大家推荐一个那就是下面这个版本,torch版本为1.12.0,cuda版本为11.6,python版本选择为python=3.9。
pip install torch==1.12.0+cu116 torchvision==0.13.0+cu116 torchaudio==0.12.0 --extra-index-url https://download.pytorch.org/whl/cu116
三、这个版本我自己在YOLOv11,YOLOv8亲自配过几次,发现确实是很好用的。其好用的原因就是在安装其他的库时,不会出现因版本问题而导致的各种疑难杂症样的报错。