tensorflow的安装(新手向)
安装步骤和我只前的一篇博客中有很大的重复,主要是cuda和cudnn的安装问题
经过验证 cuda10.1 和cudnn 7.6.5 在tensorflow2.2中可用,看其他文章中2.0 ,2.1都是正常使用
https://blog.csdn.net/beifangc/article/details/108967239
只需要将安装torch的部分换成
pip install tensorflow-gpu==2.2 #速度慢可使用镜像
同时忽略掉yolov5的下载安装部分即可
还需要注意 python版本的配合
https://docs.floydhub.com/guides/environments/
这里面有tensorflow各个版本对应的python版本要求