如何通过tensorflow指定使用特定GPU进行计算？

本文共计2181个文字，预计阅读时间需要9分钟。

TensorFlow是一款基于数据流编程（dataflow programming）的符号数学系统，广泛应用于各类机器学习（machine learning）算法的编程实现。它源于谷歌的神经网路计算库DistBelief。[1]

TensorFlow是一个基于数据流编程（dataflow programming）的符号数学系统，被广泛应用于各类机器学习（machine learning）算法的编程实现，其前身是谷歌的神经网络算法库DistBelief [1] 。
Tensorflow拥有多层级结构，可部署于各类服务器、PC终端和网页并支持GPU和TPU高性能数值计算，被广泛应用于谷歌内部的产品开发和各领域的科学研究 。

TensorFlow由谷歌人工智能团队谷歌大脑（Google Brain）开发和维护，拥有包括TensorFlow Hub、TensorFlow Lite、TensorFlow Research Cloud在内的多个项目以及各类应用程序接口（Application Programming Interface, API） 。自2015年11月9日起，TensorFlow依据阿帕奇授权协议（Apache 2.0 open source license）开放源代码 。

持续监控GPU使用情况命令：

$ watch -n 10 nvidia-smi

一、指定使用某个显卡

如果机器中有多块GPU，tensorflow会默认吃掉所有能用的显存， 如果实验室多人公用一台服务器，希望指定使用特定某块GPU。