设置可见GPU,进行多显卡深度学习训练

在深度学习中,如果一台电脑具有多个NVIDIA的GPUs,用户想要在不同的GPU上训练不同的网络,那么在程序中指定占用的GPU的id,在python中如:

import os
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"   # see issue #152
os.environ["CUDA_VISIBLE_DEVICES"]="0"

即可指定GPU id为0的GPU可见,其他的不可见,就不会占用其他GPU了。

posted @ 2017-10-01 10:01  FesianXu  阅读(123)  评论(0编辑  收藏  举报