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

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了~

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值