tensorflow2.0安装

2020-01-15  本文已影响0人  LoveToday2020

安装tensorflow2.0

1.CPU版本安装

a.确保pip版本不低于19.0

b.pip install tensorflow

2.GPU版本安装

a.检查Nvidia显卡驱动

命令行执行nvidia-smi, 确保版本不低于410.x,否则要升级

2.安装Anaconda

3.执行安装

a. conda install tensorflow-gpu==2.0.0

启动notebook工具
1.启动notebook

启动notebook命令:   jupyter notebook

2.如果是检查当前环境是否支持GPU版本

tf.test.is_gpu_available()

如果返回的结果是true则表示支持GPU版本

查看tensorflow 版本

print('Tensorflow Version:{}',format(tf.__version__))

Tensorflow Version:{} 2.0.0

上一篇 下一篇

猜你喜欢

热点阅读