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

来源:互联网 发布:淘宝售后 时间 编辑:程序博客网 时间:2024/06/07 00:12

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

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

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

阅读全文
0 0
原创粉丝点击