RTX3060+ubuntu20.04+cuda11.1+cudnn8.0.5+pytorch1.7.1+tensorflow2.4构建深度学习环境

由于2070显卡损坏,换成3060,造成了一系列影响,总共用了大概一周的时间,2个全天,重新配置深度学习环境。

问题:使用的早期版本的yolo模型,makefile文件编译不过,报错。

最直接的解决方案:使用pytorch版本的yolo,但是我之前在目前版本的yolo上作了大量工作,不想放弃。

解决过程:RTX30系列最低支持11.1,配置过程中ubuntu18.04出现了很多问题,不想一一解决了,换成了ubuntu20.04,这也造成了很多问题,ubuntu20.04的python是3.8.5,gcc是9点多。
首先尝试了cuda11.1、11.2、11.0(解决过程中因为问题太多,造成记忆混乱,后面才发现不适配11.0),都不能解决。
换成显卡2060s(同门的),结果驱动不能检测到……
换成我之前的2070,cuda10.1,完美运行(这说明我的方法没错,之前给别人用2060s装系统,也是没有跑起来。心里没数,自己之前配置2070时也遇到很多问题)

重点:仔细读编译的错误问题

关键解决方法:

  1. gcc降级为7,python使用3.7或者3.6.6
  2. 更换新版本的yolo,https://github.com/AlexeyAB/darknet

致谢:
[1]: https://blog.csdn.net/mdllll/article/details/110072342
[2]: https://blog.csdn.net/qq_39938666/article/details/86611474
[3]: https://blog.csdn.net/HaoZiHuang/article/details/109544443
[4]: http://adrai.github.io/flowchart.js/

你可能感兴趣的:(深度学习环境搭建,深度学习,人工智能,python,cuda,ubuntu)