首先 import os

然后通过 os.environ[‘CUDA_VISIBLE_DEVICES‘]=来设置用哪张卡

比如使用编号为0的卡：import os
                                     os.environ[‘CUDA_VISIBLE_DEVICES‘]=‘0‘

深度学习 多卡 python设置