yolo环境 显卡版本Cuda pytorch可能要注意的地方
·
Pytorch安装注意事项
16XX的显卡，安装cu102的版本，否则可能训练出现问题
30XX、40XX显卡，要安装cu111以上的版本，否则无法运行
显卡型号geforce 1650 适合的版本cuda10.2
torch1.10.1
# CUDA 10.2
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=10.2 -c pytorch
# CUDA 10.2
pip install torch==1.10.1+cu102 torchvision==0.11.2+cu102 torchaudio==0.10.1 -f https://download.pytorch.org/whl/cu102/torch_stable.html
ultralytics 版本选择
| 需求场景 | 推荐库版本 | 说明 |
|---|---|---|
| 使用 YOLOv8 | 8.3.57 | 稳定、经过充分测试 |
| 使用 YOLOv11 | 8.1.15 或更高 | 需最低支持该模型的版本 |
| 同时使用 v8 和 v11 | >=8.1.15 | 需手动切换模型文件 |
更多推荐
3
0- 0
已为社区贡献5条内容
所有评论(0)