这又是一个奇怪的问题,我的GPU和CPU版本的TF都是2.4.1
然后发现如果要使用TF2.4.1那么需求是安装有CUDA 11.0,而在2.3之前,还是10.1
为了兼容性装回2.3.0
pip install --upgrade tensorflow-gpu==2.3.0
pip install --upgrade tensorflow==2.3.0
装回后再看看版本
pip3 list
再次的运行写好的软件,看起来是没问题了。但是我用的是RTX30xx,会较长时间的停留在Adding visible gpu devices: 0这里