安装tensorflow2

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。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值