Data/Errors
torch.cuda.is_available()이 False일 때
neulvo
2022. 2. 22. 14:35
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의 코드를
프롬프트에 복사, 붙여넣기 해서 설치.
작동 확인 및 cuda코드의 train() 진행.
728x90