반응형
Pytorch CUDA 사용
-
[Pytorch Tip!] 파이토치 GPU 사용 설정2) Tech 2020. 9. 23. 16:53
cuda를 version에 맞게 setting 해주었다면, 사용을 해봐야겠죠? 이번시간에는 간단하게 파이토치에서 GPU를 사용가능하도록 하는 명령어를 소개하겠습니다. # pytorch gpu 사용법 import torch USE_CUDA = torch.cuda.is_available() print(USE_CUDA) device = torch.device('cuda:0' if USE_CUDA else 'cpu') print('학습을 진행하는 기기:',device) 만약 cuda 설정이 안되어있다면, 옆 이미지와 같이 cpu로 결과가 나오겠죠? 이러한 결과가 나와야 gpu사용이 가능한겁니다!! 이제, 위에서 선언한 device가 GPU로 설정되었습니다. 설정된 device에 대한 정보를 확인해보겠습니다. p..