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安装这些版本以后出现如下错误
YOLOV5 服务器pytorch环境配置踩坑问题
应该就是环境不匹配的问题吧


解决方法:找了一台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
YOLOV5 服务器pytorch环境配置踩坑问题
愁人的问题又多了,这个时候菜鸡在网上又找到了一个博主cuda11.2安装pytorch的教程,发现可以选择安装cuda11.0对应的命令
乐呵呵找到了这个
YOLOV5 服务器pytorch环境配置踩坑问题
结果就是一直装一直由于网络问题断开,装的抓狂想吐,网友推荐可以离线下载,就在这个网址找了对应的版本下载(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

上一篇:yolov5代码解读-网络架构


下一篇:coco2017与yolov5