代码是基于tensorflow1.15.4编写
显卡是3090
冲突:显卡3090需要cuda11.0+,tensorflow1.15需要cuda9.0
解决办法:
(1)只针对ubantu系统
安装nvidia-tensorflow
(2)通用解决方法
在代码最开始加如下代码:
import tensorflow.compat.v1 as tf
tf.compat.v1.disable_eager_execution()
可能还会遇到个别tensorflow2.0+和tensorflow1.0+写法不一样的地方,对应修改即可
上述文章可能部分源于网络
整理在此,只为个人记录学习过程,并分享给需要的小伙伴
如有侵权,联系即删