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