- 首先需要安装好python,可参考Python基础(2)-多版本兼容安装部署(py27,py34,py35,py36,py37,py38,py39)
- 如果需要安装虚拟环境,可参考:Python基础(1)-Miniconda+Pipenv安装部署开发环境实战详解
- 使用如下命令安装 TensorFlow 2.0
- 安装cpu版本 :pip install tensorflow
- 安装gpu版本:pip install tensorflow-gpu
- 编写如下代码验证一下机器是否支持GPU
import tensorflow as tf
print(tf.test.is_gpu_available())
a=tf.constant(2)
b=tf.constant(4)
print(a+b)
如下,表示当前机器是不支持GPU加速显卡,那就先试用cpu版本做测试学习,加法测试已经OK了
- 至此,TensorFlow 已经安装正确了