tensorflow2添加命令使用cpu训练

标签：nbsp   版本   body   compute   cells   pac   bsp   add   developer   

https://developer.nvidia.com/compute/cuda/10.0/Prod/local_installers/cuda_10.0.130_411.31_win10

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

tensorflow2添加命令使用cpu训练

标签：nbsp   版本   body   compute   cells   pac   bsp   add   developer   

原文地址：https://www.cnblogs.com/zhangdangdang/p/13842994.html