tensorflow 2.0 学习笔记 (一) 安装与初探

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

发布了161 篇原创文章 · 获赞 71 · 访问量 11万+

猜你喜欢

转载自blog.csdn.net/yxpandjay/article/details/102582863