tensorflow2.0 实现gpu和cpu切换

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

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

posted @ 2019-11-15 09:54  BigZhan  阅读(8084)  评论(1编辑  收藏  举报