yolov5训练之使用GPU

cpu跑代码慢到怀疑人生，还得是gpu啊
但是你得有gpu，还得安装cuda。。。。。。一系列操作之后才能跑

1、首先查看笔记本cuda版本，安装对应cuda。在英伟达控制面板查看即可（网上很多，自己查）

安装教程在这cuda安装
cuda向下兼容，所以安装比笔记本对应版本低的也可以，但高的不行

2、安装cuda版本的pytorch包

首先得查看torch、python的版本，在这里下载对应版本的包python库

3、安装

在安装包的路径下：
pip install 下载的包的名字

注意：先安装torch，再安装torchvision

全部完成之后，运行代码，自动使用gpu：