YOLOV5 服务器pytorch环境配置踩坑问题

YOLOV5 服务器pytorch环境配置踩坑问题

碎碎念：装这个环境可真的是累到我了，一开始在一个大服务器上跑，结果发现cuda版本只有8.0，就在pytorch官网查找对应版本torch，发现对应的torch只有1.0版本，根本没有达到requirements的1.7.0。在网上找踩坑经验 发现成功大部分都是如下配置。

torch=1.7.1+cu110
torchvision=0.8.2+cu110
但是呢，因为服务器cuda：8.0安装这些版本以后出现如下错误

应该就是环境不匹配的问题吧

---

解决方法：找了一台cuda：11.2版本的服务器，按照如下版本装
（前提是创建一个虚拟环境哦，会方便很多）

python：3.7
cuda：11.2
torch：1.7.1+cu110
torchvision：0.8.2+cu110

具体安装过程
我是在线下载的，需要根据自己的cuda版本选择命令，网址：https://pytorch.org/get-started/previous-versions/
开心的点进去后发现没有我对应的cuda版本，目前最高只到11.0

愁人的问题又多了，这个时候菜鸡在网上又找到了一个博主cuda11.2安装pytorch的教程，发现可以选择安装cuda11.0对应的命令
乐呵呵找到了这个

结果就是一直装一直由于网络问题断开，装的抓狂想吐，网友推荐可以离线下载，就在这个网址找了对应的版本下载（https://download.pytorch.org/whl/torch_stable.html）
但是需要拷到服务器好麻烦，我好懒，就在pytorch官网找找有没有别的命令，结果重点来啦！！!

下面展示一些 内联代码片。

# CUDA 11.0
pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

发现还有这命令，就在最后试了试，噢噢噢噢~很顺利就装好了对应的pytorch版本，现在YOLOV5可以跑起来了，可以开心的干饭了哈哈哈哈哈。

---

参考链接：
1.https://blog.csdn.net/qq_41251196/article/details/114108223
2.https://blog.csdn.net/yumin1058882119/article/details/107107167/?utm_medium=distribute.pc_relevant.none-task-blog-baidujs_title-5&spm=1001.2101.3001.4242
3.https://blog.csdn.net/qq_35975447/article/details/116852693