(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"
728x90
