yolov5（GPU）配置

---

一.yolov5配置（GPU）

我的电脑配置 cuda 10.0
打开CMD，输入nvcc --version

1. 复制配好的cpu环境

找到anaconda的envs文件夹里的yolov5文件夹（昨天配置好的cpu环境）复制粘贴到envs文件夹里并改名为yolov5GPU

2. 下载GPU版本的pytorch和pytorchvision

打开pytorch官网https://pytorch.org/get-started/locally/查看需要安的软件包版本

打开https://download.pytorch.org/whl/torch_stable.html下载所需的软件包（从pytorch官网下载太慢）
由于我的是cuda10.0我安的软件包是：torchvision-0.7.0+cu101-cp38-cp38-win_amd64.whl和torch-1.6.0+cu101-cp38-cp38-win_amd64.whl

下完之后发现只有跑yolov5s.pt模型没有问题但是跑yolov5x.pt会出现问题（具体问题见bug里的描述），于是我下载了
torchvision-0.8.2+cu101-cp38-cp38-win_amd64.whl
torchaudio-0.7.2-cp38-none-win_amd64.whl
torch-1.7.1+cu101-cp38-cp38-win_amd64.whl

将原来cpu安装的torch和torchvision卸载

pip uninstall torch
pip uninstall torchvision

安装gpu版的torch等一系列包

pip install torchvision-0.8.2+cu101-cp38-cp38-win_amd64.whl
pip install torchaudio-0.7.2-cp38-none-win_amd64.whl
pip install torch-1.7.1+cu101-cp38-cp38-win_amd64.whl

(参考资料：1.https://www.bilibili.com/video/BV1FK411K78w?from=search&seid=354142551239432005
2.https://blog.csdn.net/water19111213/article/details/104352503
3.https://www.dtmao.cc/news_show_576859.shtml)

3.测试pytorch和torchvision是否安装正确

import torch
flag = torch.cuda.is_available