Skip to content

Latest commit

 

History

History
95 lines (51 loc) · 1.24 KB

File metadata and controls

95 lines (51 loc) · 1.24 KB

TensorFlow2.0基础

安装Tensorflow

Anaconda下安装Tensorflow2.0

anaconda的安装方式不再赘述,首先虚拟出一个tf2.0的环境:

conda create -n tf20 python=3.6

然后激活这个环境:

conda activate tf20

使用pip来安装tf2.0:

pip install tensorflow

安装完以后输入如下命令来测试:

pip list | grep tensor

显示

tensorboard          2.0.2              
tensorflow           2.0.0              
tensorflow-estimator 2.0.1  

或者在python中输入:

import tensorflow
tensorflow.__version__
tensorflow.keras.__version__

output:

'2.0.0-alpha0'
'2.2.4-tf'

至此tensorflow2.0安装完成。

退出tf20环境:

conda deactivate

linux系统下安装Tensorflow2.1

具体参考:Centos7 安装Tensorflow2.1 GPU以及Pytorch1.3 GPU(CUDA10.1)

测试是否支持GPU

print(tf.test.is_gpu_available())

参考资料

"Anaconda下安装Tensorflow2.0"参考此博客。