赞
踩
版本不匹配可能导致的问题有:①程序出错;②gpu无法启动,只能用cpu训练。
CUDA10.X版本最高支持算力为7.5,不满足显卡8.9的算力,所以应该下载11.X的版本。可学习CUDA和显卡驱动以及pytorch版本的对应关系
因此我下载了CUDA11.0.1的版本。
根据和tensorflow-gpu的版本对应关系(官网查询)查到对应python版本和tensorflow-gpu以及cudnn的版本要求。
根据cuda版本以及yolov5的要求(requirements.txt)选择torch和torchvision版本(官网旧版本查询)(新版本)。再去下载相应的包,需要注意和cuda、python的对应,下载地址附在最后。
这时可能就会出现版本冲突问题。
冲突的体现方式是pip一个安装包,安装成功之后,会发现命令行中出现ERROR,表示虽然安装上了,但是版本不匹配。
也有可能是pycharm中提示错误。
以下为部分错误信息集结:
综合这些错误信息与yolov5的要求,可以降低matplotlib、pandas、ultralytics的版本。
唯一的矛盾就在于tensorflow-gpu的版本不能降低只能升级,但是那样就不满足第二个要求中和cuda等的匹配了,由于官网给出的是已经测试过没有问题的版本,所以没有测试过的只能自己尝试一下了。
于是我选择了tensorflow-gpu==2.9.0的版本,没想到成功了!!!
当然也可以选择试一下降低tensorflow-gpu的版本,有可能官网给出的是最高可以容纳的版本。
模块名称 | 版本号 |
---|---|
cuda | 11.0.1 |
cudnn | 8.0 |
torch | 1.7.1 |
torchvision | 0.8.2 |
tensorflow-gpu | 2.9.0 |
其他 | 满足要求即可 |
沾沾喜气~
模块名称 | 方法 | 教程 |
---|---|---|
cuda | 官网下载 | 参考教程 |
cudnn | 同上 | 参考教程 |
torch(pytorch) | 先下载安装包whl,再pip install路径 | 参考教程 |
torchvision | 同上 | 参考教程 |
tensorflow-gpu | pip install tensorflow-gpu==2.9.0 镜像地址 | |
其他 | pip install即可,或者统一根据pycharm的提示 |
镜像地址使用方法:
方法一:pip install 安装包名字 -i http://pypi.doubanio.com/simple/ --trusted-host pypi.doubanio.com //豆瓣镜像网站
方法二:pip install 安装包名字 -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com //豆瓣
方法三:pip install 安装包名字 -i https://pypi.tuna.tsinghua.edu.cn/simple/ --trusted-host pypi.tuna.tsinghua.edu.cn //清华大学
当出现版本不匹配的错误之后,如何卸载相应的模块呢?
这个教程是比较靠谱的。
之前看到一个教程说让带nivida的都删掉,结果我把显卡驱动也删了,屏幕显示都出现问题(气!)。
搞得我很头大,也是涨了不少知识。
模块名称 | 方法 |
---|---|
cuda、cudnn | 卸载NIVIDIA相关程序,把相应版本号的删掉即可 |
tensorflow-gpu以及torch相关 | pip uninstall即可 |
python | 配置虚拟环境的时候设置的,可以重建虚拟环境 |
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。