step3 pytorch安装
首先进入到pytorch官网:pytorch官网
为了避免不兼容,我们下载历史版本的pytorch,暂不使用最新版本的。
然后我们需要根据自己的电脑来选择相应版本的pytorch。
我们在终端输入:
nvidia-smi
可以看到我的这个电脑版本的显卡是11.6的所以我们在官网上下载一个更老的版本(为了防止无法训练)。
这个包有2.4G,让我们耐心等待一下吧。
终于下载好了!
step4 安装ultralytics(yolov8)
具体可以参考官方文档: