欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Ubuntu16.04使用Conda安装Tensorflow(GPU版本)

程序员文章站 2022-06-05 20:42:24
...

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的版本匹配问题,参考博客:

https://blog.csdn.net/MahoneSun/article/details/80809042

附Tensorflow清华镜像源链接

4.安装Spyder

tensorflow的安装到上一步就算是完成了。但是由于我们是在虚拟环境里安装的,所以直接在spyder编译器里面是无法导入的。因此需要在这个环境里再安装一个spyder。

conda install spyder

对于vscode来说,可以直接在底栏选择编译器的环境,因此可以省去这一步。

5.后悔药

如果安装失败,或者想重新安装的话,可以删除掉这个环境从头再来:

conda env remove -n tf

现在就可以愉快的使用tensorflow了~