torch的版本跟cuda不对应导致的数据集训练过慢

在服务器上训练yolov5的数据集，2080ti显卡训练一天发现只训练了十几轮。在\runs\train\exp中打开result可看训练的轮数。

查看显卡使用情况 ，发现显卡未使用

nvidia-smi

猜测可能是torch的版本跟cuda不对应导致的，下载新的torch。可以pip安装，但是太慢，网上有很多教程，这里介绍其中一种。可参考https://blog.csdn.net/thy0000/article/details/122652349

首先去官网下载对应的.whl文件，由于我是在服务器训练，所以下载的是linux版本。把下载好的文件放到文件夹里，我这里直接放进了工程文件夹，然后cd到文件夹所在路径

先安装torch，再安装torchvision。安装完再进行训练发现速度提升，查看显卡也在使用。