import os os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID" # Arrange GPU devices starting from 0 os.environ["CUDA_VISIBLE_DEVICES"]= "0,1" # 0번 과 1번 GPU를 사용, Set the GPU 2 to use 위의 코드를 실행할 코드 맨 위에 적용하여 실행하면 됨.