为tensorflow指定GPU

为tensorflow指定GPU，原因是，默认创建session时，会将所有显存占满，发现有人在用的时候，就会session不能创建而报错。 首先nvidia-smi查看显卡的编号，最左边一列，看看哪个空的

2.在终端设置使用的GPU

如果用方法一，虽然方便，但有的时候还是需要指定其他的GPU，这时可以这样，例如

CUDA_VISIBLE_DEVICES=2 python test.py

这样就只会使用序号为2的GPU

3.在程序中指定使用的GPU

import os

os.environ["CUDA_VISIBLE_DEVICES"]=‘2’

这里仅做一下记录，以防清华镜像不稳定。 linux，GPU， py2.7

pip install \
  -i https://pypi.tuna.tsinghua.edu.cn/simple/ \
  https://mirrors.tuna.tsinghua.edu.cn/tensorflow/linux/gpu/tensorflow_gpu-1.4.0-cp27-none-linux_x86_64.whl