tensorflow2添加命令使用cpu训练

Posted 2020-12-20 zhangdangdang

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了tensorflow2添加命令使用cpu训练相关的知识，希望对你有一定的参考价值。

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了

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

以上是关于tensorflow2添加命令使用cpu训练的主要内容，如果未能解决你的问题，请参考以下文章