cuda11.1的ultralytics环境配置

最新推荐文章于 2025-02-26 00:37:22 发布

小枕巾

最新推荐文章于 2025-02-26 00:37:22 发布

阅读量488

点赞数 10

文章标签：深度学习人工智能

本文链接：https://blog.csdn.net/m0_69761150/article/details/144342815

版权

annaconda 创建环境ul，python=3.9
安装pytorch（https://download.pytorch.org/whl/torch_stable.html），我的cuda是11.1，安装了torch-1.10.1+cu111-cp39-linux_x86_64.whl和对应的torchvision-0.11.1+cu111
安装ultralytics库项目首页 - ultralytics:ultralytics - 提供 YOLOv8 模型，用于目标检测、图像分割、姿态估计和图像分类，适合机器学习和计算机视觉领域的开发者。 - GitCode
```
pip install ultralytics
```
在ul环境中检查环境中的包（pip list）是否符合要求（ultralytics/pyproject.toml at main · ultralytics/ultralytics · GitHub）
下载源码
在ultralytics-main创建train.py
用vscode打开下载的项目文件夹ultralytics，在vscode中的终端开始训练
```
python train.py
```