0.写在前面
这是第一次使用Markdown语言,也是第一次写博客,谨以此博客正式开启深度学习之路。
前两天为了安装Tensorflow各种折腾,最后终于在别人的帮助下熟悉了这一整套流程,特此记录下来以便日后查阅。
1.使用Conda新建环境
conda create -n tf python=3.6
这里新建了一个名为tf的环境,指定Python版本为3.6。
2.在新环境里升级pip
首先激活新环境:
source activate tf
然后升级pip(如果pip版本过低后面可能会报错):
pip install --upgrade pip
3.从清华镜像源安装Tensorflow
pip install tensorflow-gpu==1.10.1 -i https://pypi.tuna.tsinghua.edu.cn/simple/
本人的环境为Ubuntu16.04LTS+CUDA 9.0+cuDNN 7.1.3, 所以这里指定tensorflow版本为1.10.1(更高版本的tensorflow应该也可以,但是有可能会出现问题)。关于CUDA、cuDNN和tensorflow的版本匹配问题,参考博客:
4.安装Spyder
tensorflow的安装到上一步就算是完成了。但是由于我们是在虚拟环境里安装的,所以直接在spyder编译器里面是无法导入的。因此需要在这个环境里再安装一个spyder。
conda install spyder
对于vscode来说,可以直接在底栏选择编译器的环境,因此可以省去这一步。
5.后悔药
如果安装失败,或者想重新安装的话,可以删除掉这个环境从头再来:
conda env remove -n tf
现在就可以愉快的使用tensorflow了~