tensorflow2.4 问题记录

1、训练时出现:could not successfully open dynamic XXX.dll,导致不能使用GPU训练,而使用CPU训练

  因为RTX3060 对应 CUDA11.1 对应 tensorflow 2.4 ,而自带的动态库是cusolver64_11.dll,需要cusolver64_10.dll,从网上下载对应的动态库,放到当前工作目录。

 

2、keras 库问题

  tensorflow2.4.0 已经自带keras了,以前加载keras的代码只要在前面加tensorflow,如

import tensorflow.keras.backend as K

 

posted on 2022-02-11 11:26  WenJXUST  阅读(92)  评论(0)    收藏  举报