bert4keras 大佬！30系显卡发布这么久了，什么时候更新支持CUDA11.+，TF2.6+的版本啊。

大佬！30系显卡发布这么久了，什么时候更新支持CUDA11.+，TF2.6+的版本啊。

Open SonGohan757 opened this issue 3 years ago • 4 comments

如题

Nov 04 '21 02:11 SonGohan757

1、tf 2.x，用TF_KERAS=1，勉强能用；

2、30系列，包括更高级的A100，最佳的tf版本是1.15，这是nvidia自己说的：https://developer.nvidia.com/zh-cn/blog/accelerating-tensorflow-on-a100-gpus/

Nov 04 '21 02:11 bojone

简单来说，就是nvidia自己维护的tf分支，才是对gpu支持最好的tf，而nvidia自己维护的tf有1.15并且支持cuda11并且还是对新显卡支持最好的tf

Nov 04 '21 02:11 bojone

可以试试docker 真的香

Jan 06 '22 14:01 MichaelMachine

可以试试docker 真的香

你好，docker的话是起始镜像为tensorflow-gpu 然后镜像里面pip Install bert4keras 吗，刚试了keras 版本有冲突，tensorflow依赖版本keras2.8，bert4keras 依赖版本2.3，请问这个版本问题咋解决，还是我docker使用有问题，谢谢

May 12 '22 03:05 kingglory