主要内容,快速安装运行依赖库使用官方源需要较好的网络环境,适用于nvidia独显的电脑
前期准备
- 良好的网络环境,最好用科学上网
- 安装好miniconda或者conda
- 安装了显卡驱动,查看自己的cuda版本,只需要知道大版本号是11还是12就可以
环境搭建
创建一个conda环境
conda create -n [环境名称] python=[python版本,至少3.8]
示例
conda create -n torch python=3.8
激活环境安装pytorch,打开pytorch主页,选择合适的版本然后复制命令并运行
安装其它依赖
conda install conda-forge::ultralytics
补充:还需要再安装thop包,否则不显示gflops
pip install thop
运行detect.py测试