pip安装指南https://tensorflow.google.cn/install/pip?hl=zh_cn
tensorflow相关问题解决方法https://www.tensorflow.org/install/errors
pip install tensorflow安装之后碰到一些小问题,import tensorflow 报错 dll加载失败,还需要注意python版本和tensorflow版本的兼容
需要安装vc运行库
如果需要使用gpu加速的tensorflow还需要安装cuda,未安装会报错“Could not load dynamic library 'cudart64_101.dll'; dlerror: cudart64_101.dll not found” 之类的,提示找不到dll
以及安装cudnn,具体参考https://www.cnblogs.com/baby123/p/13035830.html
下载cudnn遇到个坑,下载下来的压缩包只有几十kb,怎么解压都失败,需要开vpn再点击下载链接压缩包大小是七百多兆,可以迅雷接管下载。
跑的时候发现cuda(11.1)和cudnn(8)的版本太高了,tensorflow(2.3.1)对应的是cuda(10.1)和cudnn(7),网上有人说直接改dll的名字,我直接将dll名称都改了后就能跑了,
但是发现tf.cast函数将整数张量转为bool类型的张量得不到正确结果,禁用gpu之后再cpu上跑就能得到正确结果。不想重装了,等待tensorflow可以支持cuda11。