摘要:
1、训练时出现:could not successfully open dynamic XXX.dll,导致不能使用GPU训练,而使用CPU训练 因为RTX3060 对应 CUDA11.1 对应 tensorflow 2.4 ,而自带的动态库是cusolver64_11.dll,需要cusolver
阅读全文
posted @ 2022-02-11 11:26
WenJXUST
阅读(58)
推荐(0)
编辑
摘要:
1、to_float,to_int32 # has_min = tf.to_float(tf.reduce_any(pos_num_neg_mask)) has_min = tf.cast(tf.reduce_any(pos_num_neg_mask), tf.float32) 2、tf.log #
阅读全文
posted @ 2022-02-11 11:15
WenJXUST
阅读(233)
推荐(0)
编辑