1. 程式人生 > >設定可見GPU,進行多顯示卡深度學習訓練

設定可見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了。