YOLOv5使用GPU训练
前提:yolov5在CPU训练可以进行后,想改用GPU提速遇到问题。原因:自己的torch下的是CPU版本。方法:重新创建一个解释器。step1:conda remove -n yolo --allstep2:conda create -n yolov5 python=3.8step3: 激活解释器,去PyTorch官网下载对应GPU版本的torch(用pip下载)step4:把yolov5文件下
·
前提: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
更多推荐
已为社区贡献1条内容
所有评论(0)