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
 
                    
                 
 
                
            
         浙公网安备 33010602011771号
浙公网安备 33010602011771号