10月1日,tf 2.0正式版出来了。几点重要改动:
1. 默认Eager Execution模式。静态图debug多蛋疼写过的都知道。pytorch工业部署并不方便
2. 统一接口。以前的tf文档和接口乱的不行。现在统一用keras
3. 新增TensorFlow Datasets。为包含大量数据类型的大型数据集提供了标准接口
安装步骤如下:
先装好anaconda。依次敲命令:
conda create -n tf2c python=3.6.7 (似乎python=3.7也可以)
conda activate tf2c
conda install cudatoolkit=10.0 cudnn=7.3 (然而运行卷积层报错,调查后发现要求tf2要求cudnn>7.4.1)
##conda install -c anaconda cudatoolkit (这句可以不要)
conda install -c anaconda cudnn
pip install tensorflow-gpu==2.0.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
可以创建一段代码试试安装是否成功:
import tensorflow as tf
version = tf.__version__
gpu_ok = tf.test.is_gpu_available()
print("tf version:",version,"\nuse GPU",gpu_ok)
贴两个教程:
https://tf.wiki/zh/basic/installation.html
https://blog.csdn.net/abc13526222160/article/details/101938410