在window和linux上安装tensorflow-gpu版–踩坑

  • Post author:
  • Post category:linux


在window和linux上安装tensorflow-gpu版



安装教程

先强调一下 驱动 cuda cudnn tensorflow 版本对应问题一定要匹配 一定! 否则后面都是麻烦

具体匹配详情


https://tensorflow.google.cn/install/source


注:我安的是cuda10 cudnn7.5 tensorflow1.130rc0 之前按7.4正常运行一段时间报 cudnn failed to initialize…错 所以重装到7.5了,cudnn的卸载也非常简单,只需删除libcudnn与cudnn相关文件即可

下面这个链接是在win10上基于anaconda安装(比较推荐anaconda安装这算是一个独立的环境)


https://blog.csdn.net/ebzxw/article/details/80701613


安装时遇到的问题:

1.环境变量cuda的要加,cudnn的也要加,之后重启使环境变量生效

2.SSL error代理关闭之后,安装openssl就可以了

这个链接是在Ubuntu上安装tensorflow-gpu的链接


https://blog.csdn.net/macunshi/article/details/84638675

比较值得注意的地方:

1.驱动-cuda-cudnn-tensorflow-gpu 对应好版本 我的是(驱动410.48-cuda10.0-cudnn7-tensorflow1.13某个版本, win10上是先卸载驱动然后装cuda9.0中自带的驱动, 再解压cudnn按步骤放入cuda文件夹下的一些文件夹)

2.安装错误使用sudo apt-get –purge remove cuda-*-10-0卸载cuda,sudo apt-get remove –purge nvidia-*卸载驱动

3.有些cudnn.h不在/usr/local/cuda/include/目录下 ,而在/usr/include/ 下

19.04.13

本来一切都很正常,跑了几个tensorflow的程序,但今天突然就报错

Failed to get convolution algorithm. cudnn failed to initialize…

问题原因 tensorflow版本与cudnn版本不匹配

解决方法 tensorflow降级或者cudnn安装对应cuda的最新版本

但这个错真的莫名其妙 突然就出现了,之前还好好的



版权声明:本文为xzy528521717原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。