昨天把GPU版本的tf2.0 安装成功之后，现在所有的代码运行居然都在gpu上跑了，
并且在对gpu使用情况没有限制的条件下，既然gpu内存跑满了，代码就崩了
怎么样才能随心所欲的指定代码是在cpu还是gpu呢
首先若不加任何配置情况下，是默认使用gpu的，
加上下面这句代码就使用cpu了
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
昨天把GPU版本的tf2.0 安装成功之后，现在所有的代码运行居然都在gpu上跑了，
并且在对gpu使用情况没有限制的条件下，既然gpu内存跑满了，代码就崩了
怎么样才能随心所欲的指定代码是在cpu还是gpu呢
首先若不加任何配置情况下，是默认使用gpu的，
加上下面这句代码就使用cpu了
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"