[Torch] pyTorch 자주 쓰는 명령어 정리

m반쩨 GPU device 사용

improt os

# 1번째 device만 사용
os.environ["CUDA_VISIBLE_DEVICES"]= "1"
# 2, 3번째 device 사용
os.environ["CUDA_VISIBLE_DEVICES"]= "2,3"


# 혹은 terminal에서 export 환경변수로 선언해 사용
# export CUDA_VISIBLE_DEVICES=1

GPU 사용 가능여부 확인

torch.cuda.is_available()

현재 사용중인 device 번호 확인

device slot 순서와 상관없이 기본적으로 0번부터 할당 (즉 장치가 1개라면 0으로 출력)

torch.cuda.current_device()

현재 사용중인 device 개수 확인

torch.cuda.device_count()