需要将指定GPU代码放在程序段最开始的部位。
os.environ["CUDA_VISIBLE_DEVICES"] = '1'
# (如果你想同时调用两块GPU的话)
os.environ["CUDA_VISIBLE_DEVICES"] = '1,2'
需要将指定GPU代码放在程序段最开始的部位。
os.environ["CUDA_VISIBLE_DEVICES"] = '1'
# (如果你想同时调用两块GPU的话)
os.environ["CUDA_VISIBLE_DEVICES"] = '1,2'