运行tensorflow python程序,限制对GPU和CPU的占用操作

一般情况下，运行tensorflow时，默认会占用可以看见的所有GPU，那么就会导致其它用户或程序无GPU可用，那么就需要限制程序对GPU的占用。并且，一般我们的程序也用不了所有的GPU资源，只是强行霸占着，大部分资源都不会用到，也不会提升运行速度。使用nvidia-smi可以查看本机的GPU使用情况，如下图，这里可以看出，本机的GPU型号是K80，共有两个K80，四块可用。这里表示只有GPU 0和1对程序可见，因此也就限制了程序只能用GPU 0和1同样，也可以在代码里指定如果想只用CPU，不用CPU来运行程序，可以用如下命令：或者是前面是对GPU的限制，那如果不用GPU，只用CPU呢？如何限制对CPU的使用呢？