
GPU 번호 할당하기
Coding/기타
2023. 5. 22. 10:52
(23.05.22) 명령어로 gpu 번호 할당하기 예를 들어 7번에 할당하고자 한다면 아래와 같이 적어볼 수 있다. CUDA_VISIBLE_DEVICES=7 python main.py 그런데, nvtop 에서 보이는 순서와 CUDA_VISIBLE_DEVICE에서 보이는 순서가 다를 수 있다. 그럴땐, 아래와 같이 CUDA_DEVICE_ORDER=PCI_BUS_ID 명령어를 앞에 추가해주어야한다. CUDA_DEVICE_ORDER=PCI_BUS_ID CUDA_VISIBLE_DEVICES=7 python main.py 코드에서 gpu 번호 할당하기 os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"] = "7"