本文介绍TensorFlow 2.0的安装,TensorFlow 2.0兼容64位Python版本
- Python 3.5—3.7
- Python 2.7(windows平台不支持)
本文使用3.7
TensorFlow 2.0分为CPU和GPU两个版本,其中GPU版本的TensorFlow可以利用NVIDIA GPU强大的计算加速能力,使Tensorflow的运行更为高效,尤其是可以成倍提升模型训练的速度。
1、tensorflow-CPU安装
命令语句:pip install tensorflow
注:pip版本不低于19.0
pip是一个现成的,通用的Python包管理工具,提供了对Python包的查找、下载、安装、卸载的功能。
2、tensorflow-GPU安装
tensorflow对NVIDIA显卡的支持较为完备。AMD的显卡也开始对tensorflow提供支持。
注:(1)NVIDIA显卡要求其CUDA Compute Capability不低于3.0
算力参考博文
算力参考:官网
(2)NVIDIA驱动程序需410.x或更高版本
命令语句:nvidia-smi
如果遇到如下错误:‘nvidia-smi’ 不是内部或外部命令,也不是可运行的程序
则需要更改命令路径:cd C:\Program Files\NVIDIA Corporation\NVSMI
安装tensorflow 2.0的命令行语句:conda install tensorflow-gpu==2.0.0
最后在Jupyter NoteBook中检查版本:
import tensorflow as tf
tf.test.is_gpu_available()
返回为:true