设置可见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了。