Tensorflow 极简安装教程

安装TensorFlow花了好久是时间和精力,被网上的教程支配,最后才发现原来只需要一条指令就可以安装。

1 安装anaconda

anaconda的安装比较简单,没有那么多坑,网上找一个安装教程跟着做就好,这里不做赘述,放上一个链接。
链接: https://blog.csdn.net/ychgyyn/article/details/82119201

2 TensorFlow安装

准备工作

现在anaconda里给TensorFlow分配一个位置,方便管理。

conda info --envs

用来查看环境变量,看看有那些环境,我这里已经有TensorFlow-gpu了,没创建过是没有的。
查看环境变量

conda create -n tensorflow python=3.6

创建TensorFlow环境

activate tensorflow

激活TensorFlow环境,这个时候在查看环境应该可以看到TensorFlow后面有一个*,说明TensorFlow环境已经激活,进行下一步安装吧。

(1) CPU版本

在Anaconda prompt 里面执行

pip install tensorflow

安装的是最新版的TensorFlow。

(2) GPU版本

网上很多让你找cuda和cudnn的对应的版本,其实不需要。
在Anaconda prompt 里面执行

conda install tensorflow-gpu==2.0.0

2.0.0是版本号,可以自己改成别的,比如1.8.0,conda会自己安装对应的cuda和cudnn

3 测试

在Anaconda prompt 里面执行

python

进入python,然后

import tensorflow as tf

如果能够导入,并且没有报错就好,有警告可以忽视。
测试
然后输入

tf.Session()

在这里插入图片描述
跳出一堆东西,是关于GPU的,说明GPU调用成功。
TensorFlow-gpu版安装成功

其他

这里附上cuDNN和CUDA的对应的版本给大家参考

链接: https://www.jianshu.com/p/464fefb5c5d8.

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值