pytorch GPU 설정 주피터노트북

1. GPU 사용량 확인 watch -n -1 nvidia-smi 1초마다 GPU 사용량 확인 2. pytorch CUDA SETTING import os import torch os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"] = "GPU 넘버" ex) GPU 0,1,2,3 중에 1만 사용 os.environ["CUDA_VISIBLE_DEVICES"] = "1" ex)GPU 0,1,2,3 다 사용 os.environ["CUDA_VISIBLE_DEVICES"] = "GPU 넘버" 3. cuda setting 확인 device = torch.device("cpu") if torch.cuda.is_availabl..
개발/Python 2023. 1. 17. 16:32
1