YOLOv5使用GPU训练_yolov5怎么用gpu训练

前提：yolov5在CPU训练可以进行后，想改用GPU提速遇到问题。

原因：自己的torch下的是CPU版本。

方法：重新创建一个解释器。（以下一切在命令提示符窗口运行）

step1：conda remove -n yolo --all

step2：conda create -n yolov5 python=3.8

step3: 激活解释器，去PyTorch官网下载对应GPU版本的torch（用pip下载）

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

 step4：验证

python
#输入库
import torch
#查看版本
print(torch.__version__)
#查看gpu是否可用
torch.cuda.is_available()
#返回设备gpu个数
torch.cuda.device_count()
#退出python
quit()

step4：把yolov5文件下的requestments.txt文件里的torch、torchvirsion注释掉

step5：cd到对应位置，按要求安装库

d:
cd D:\WXR\yolov5-master
pip install -r requirements.txt