Ubuntu16.0.4 安装tensorflow

需要从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版本

发布了142 篇原创文章 · 获赞 53 · 访问量 20万+

猜你喜欢

转载自blog.csdn.net/qq_23301703/article/details/90343946
今日推荐