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
'Data > Errors' 카테고리의 다른 글
| [Vs Code] conda 가상 환경 변경 안 될 때 (0) | 2022.09.26 |
|---|---|
| Vue 머스태시(mustache) 랜더링 안될 때 (0) | 2022.07.19 |
| import cv2 안될 때 (0) | 2022.04.05 |
| [Errno 32] Broken pipe (0) | 2022.02.28 |
| Conda 가상 환경 생성, Kernel 추가하기 (0) | 2022.02.22 |