YOLOv8或v11使用那个torch,cuda版本好

一、大家如果使用过不同版本的yolo都会遇到一个问题,那就是在pytorch官网上有各种各样的torch和cuda的版本,到底是去选那个好。

下面的链接是pytorch官网链接。

Previous PyTorch VersionsAccess and install previous PyTorch versions, including binaries and instructions for all platforms.https://pytorch.org/get-started/previous-versions/

二、我给大家推荐一个那就是下面这个版本,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亲自配过几次,发现确实是很好用的。其好用的原因就是在安装其他的库时,不会出现因版本问题而导致的各种疑难杂症样的报错。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值