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的对应的版本给大家参考