tensorflow2.0 实现gpu和cpu切换

昨天把GPU版本的tf2.0 安装成功之后，现在所有的代码运行居然都在gpu上跑了，

并且在对gpu使用情况没有限制的条件下，既然gpu内存跑满了，代码就崩了

怎么样才能随心所欲的指定代码是在cpu还是gpu呢

首先若不加任何配置情况下，是默认使用gpu的，

加上下面这句代码就使用cpu了

os.environ["CUDA_VISIBLE_DEVICES"] = "-1"