다음 코드를 실행하면 GPU를 사용할 수 있도록 설정해 줍니다. 하지만 책에서 확인할 수 있듯이 해당 코드는 GPU 사용만을 위한 것은 아니므로 GPU 사용을 위해 코드를 따로 수정할 필요는 없습니다.
import torch device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu') # 혹은 cuda = True if torch.cuda.is_available() else False
▲ 그림 A-11 GPU 사용 설정