需要从win10转战服务器linux tf跑模型,安装了几个小时终于搞定。
遇到的问题。
1.cudnn与cuda版本 不匹配,师姐装的是cuda8.0与cudnn7.0刚好这两个哪个版本的tensorflow都不匹配,试了下莽夫直接
pip install tensorflow-gpu 可以安装成功运行报错,错误为不匹配。
2.于是前端时间学习了虚拟环境,conda大法好于是转向conda一键安装 conda install tensorflow-gpu 无需担心cudnn 与cuda版本只需要你安装好了nvidia驱动安装步骤
conda create -n tf
conda activate tf
conda install tensorflow-gpu
第一句代码创建虚拟环境
第二句代码进入虚拟环境
第三句代码一键安装
3.成功安装以为万事大吉 ,不 没这么简单 运行报错驱动与运行环境不匹配
解决方案:
1)装最新驱动
2)降低版本
当然选择2)
于是查看当前有哪些虚拟环境
conda info -e
然后新建一个虚拟环境来搞
conda create -n tf_new
conda activate tf_new
conda install tensorflow-gpu==1.9.0
搞定收工
中途还遇到小问题无法导入imsave 是scipy 与 pillow问题 版本还必须对应
pillow 6.0.0
scipy 1.21.1
如果scipy1.3则会出问题
附上驱动对应图
网址
另外 查看当前 conda 安装的是cuda几点0 在使用
conda install 的时候会有包名单 其中cudatoolkit即是cuda版本