GPU에 맞는 CUDA를 설치하고 cuDNN kit 까지 잘 설치하고 GPU 디바이스가 설치된 것이 확인됐는데도 torch.cuda.is_available()이 False라면 torch 버젼을 확인하고 다시 깔아주자. CUDA 버젼 확인 내 경우, torch version이 1.10.2+cpu로 되어있어서 gpu를 못잡았더라. Pytorch uinstall PyTorch An open source machine learning framework that accelerates the path from research prototyping to production deployment. pytorch.org 파이토치 사이트 방문 후 맞는 사양을 선택하고 Run this command의 코드를 프롬프트에 복사,..
