安装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
网友评论