安装 Tensorflow 2.1.0 版本
- 本次主要介绍如何简要的安装 Tensorflow 2.1.0 的 cpu 以及 gpu 版本。
一. 安装 cpu 版
- 在安装 tensorflow 之前,你要确保自己安装了 anaconda 。
- 在 anaconda 里面,我们只需要输入:
就可以成功安装 tensorflow 了。conda install tensorflow==2.1.0
二. 安装 gpu 版
- 安装 gpu 版,与安装 cpu 版差不多。
- 我们只需要在 cmd 里面输入:
conda install tensorflow-gpu==2.1.0
- 这样 anaconda 就会安装 tensorflow gpu 版本以及附属的依赖包了。
- 安装完成后,我们只要在 python 中输入:
就可以查看 tensorflow 的版本以及是否支持 GPU 了。>>> import tensorflow as tf >>> tf.__version__ >>> tf.config.list_physical_devices('GPU')
- 如果 tf.config.list_physical_devices(‘GPU’) 输出为 空列表,则不支持 GPU ,如果有输出则支持 GPU。
三. 期间问题
- 安装的 tensorflow-gpu 的版本为:2.1.0,打印的版本却为:1.15.2,而且无法使用 GPU。
1. 问题源头
- 输出命令:pip list
发现里面有 tensorflow 1.15.2 版本。
- 初步定位问题为:pip 包和 conda 包发生冲突。
2. 解决方法
- 卸载 pip 里面的 tensorflow 1.15.2 版本。
- 使用命令:pip uninstall tensorflow
- 卸载完成后,问题便解决了。
四. 结语
- 如果其中有不对的地方或是需要改进的地方,请指正。
- 联系方式 email:lwl510ll@163.com 。
- 本文到这就结束了,谢谢大家观看。