파이썬 GPU 선택하여 사용하기

코드 내에서 선택하기

import os

os.environ["CUDA_DEVICE_ORDER"]='PCI_BUS_ID'
os.environ["CUDA_VISIBLE_DEVICES"]='0,1'	# 0,1번 GPU 사용하기

터미널에서 파이썬 실행시 선택하기

CUDA_VISIBLE_DEVICES=0,1 python3 run.py