ylab 2022. 8. 14. 21:58

import torch

#토치 버전 확인

torch.__version__

#GPU 사용 가능 확인

torch.cuda.is_available()