tensorflow2.0安装以及教程

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/hajungong007/article/details/90140999

安装

TensorFlow 2.0依赖cuda10，但是服务器或者本地已经安装了cuda9，可以通过anaconda单独创建一个tensorflow2.0的工作环境，配置如下：

• 安装anaconda3.6或者其他版本
• 创建一个环境，用来安装tensorflow2.0以及相关的python packages.

#加上cudatoolkit and cudnn这样不论你的机器原来装的是否合适，都可以用tensorflow2.0gpu版本
conda create -n tf2 python=3.6 

• 激活创建的环境，安装tensorflow2.0

#激活环境
source activate tf2
#对于GPU版的安装，为保证不论原始linux环境是什么样的都可以安装，如果原系充已经符合，以下命令不用执行
conda install cudatoolkit=10 cudnn=7.3
#正式安装tensorflow2.0
pip install tensorflow-gpu==2.0.0-alpha0

• 测试是否安装成功

#输入python，进入python环境
import tensorflow as tf
#查看tensorflow版本
print(tf.__version__)
#输出'2.0.0-alpha0'
#测试GPU能否调用,先查看显卡使用情况
import os 
os.system("nvidia-smi")
#调用显卡
@tf.function
def f():
	pass
f()
#这时会打印好多日志，我电脑上还有warning，感觉不影响
#再次查询显卡
os.system("nvidia-smi")

教程

https://zhuanlan.zhihu.com/p/59507137

ref: https://blog.csdn.net/u011119817/article/details/88309256