项目场景:
给Linux Ubuntu18.04 安装Tensorflow+Jupyter
问题描述
按照网上最火的教程,自己下载安装cuda和cudnn,各种bug报错,最后搞得一团糟。
最后把Ubuntu直接卸载重装,重新配置
Anaconda+Tensorflow-GPU:
在ubuntu中,想要安装tensorflow-GPU,有两种方式:
- gpu-driver+cuda+cudnn+tensorflow
http://t.csdn.cn/hoIRQ - gpu-driver+anaconda+tensorflow (本人采用这种)
http://t.csdn.cn/a4PZa
http://t.csdn.cn/ej3sC
Jupyter:
安装好Anaconda后,除了base以外,建立一个虚拟环境TF115GPU,可实现tensorflow-GPU的运行。
但是基于终端的编译环境很不友好,所以希望装一个Jupyter。
一开始使用 http://t.csdn.cn/IleYU 中的方法
sudo apt install Jupyter-core
sudo apt install Jupyter-notebook
但安装并运行jupyter后,提示 No module named tensorflow
研究发现这是因为这个jupyter调用的python环境位于usr/bin中,而之前安装tensorflow-GPU的环境是在anaconda里,位于home/用户名/anaconda3
所以要安装与anaconda匹配的Jupyter,而非通用版本的Jupyter。
教程:http://t.csdn.cn/g5wJx
另外,这种方法安装的jupyter只是针对该环境(TF115GPU)的,如果想在其他环境(如base)使用Jupyter,还需再安装。
测试:
https://blog.csdn.net/weixin_44035915/article/details/110550730
用tensorflow-GPU版本大概10s内可以跑完。