Pytorch 특정 GPU만 할당 및 사용
2022. 4. 24. 15:09ㆍ프로그래밍/Pytorch
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위의 코드를 실행할 코드 맨 위에 적용하여 실행하면 됨.