为tensorflow指定GPU,原因是,默认创建session时,会将所有显存占满,发现有人在用的时候,就会session不能创建而报错。 首先nvidia-smi查看显卡的编号,最左边一列,看看哪个空的
如果用方法一,虽然方便,但有的时候还是需要指定其他的GPU,这时可以这样,例如
CUDA_VISIBLE_DEVICES=2 python test.py
这样就只会使用序号为2的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