当你安装上gpu版本的tensorflow2时,默认情况下使用GPU,当你想要切换回CPU运行程序时,
你需要的代码开头添加一句代码
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
设置使用GPU
physical_devices = tf.config.experimental.list_physical_devices('GPU')
if len(physical_devices) > 0:
tf.config.experimental.set_memory_growth(physical_devices[0],True)