前言
踩坑系列
正文
实习来了,要求下载TensorFlow,花了我一天的时间,太坑了
网络上安装教程很多了,推荐一个我安装成功比较简单的
TensorFlow安装视频教程
以下为以该视频安装为准的踩坑记录
注意点一
注意自己的电脑支不支持NVIDIA的显卡,自行选择安装gpu版本
注意点二
安好后先不要退出,输入命令conda list看一下里面有无TensorFlow模块,我一开始有,但是import tensorflow as tf出现了模块failed,参考以下解决方案。
如果出现,DLL load failed,和我一样,点击下载
Visual C++ 可再发行软件包然后重启解决问题
注意点三
安装过程中我还出现的问题是,下载完Visual C++ 可再发行软件包,import tensorflow 后,出现warning:“Python环境中HDF5报错:HDF5 library version mismatched error”
输入以下后解决问题
pip uninstall h5py
pip install h5py
注意点四
以上做好后就可以继续验证了,输入tf.__version__
出现版本号就即可。
然后在parcharm中验证
验证代码最好选择这个
import tensorflow as tf
tensorflow_version = tf.__version__
gpu_available = tf.test.is_gpu_available()
print("tensorflow version:", tensorflow_version, "\tGPU available:", gpu_available)
a = tf.constant([1.0, 2.0], name="a")
b = tf.constant([1.0, 2.0], name="b")
result = tf.add(a, b, name="add")
print(result)
出现视频里说的这一条就正确安装了
关于gpu版本的说明https://blog.csdn.net/weixin_44170512/article/details/103990592