GPU2 Segmentation fault (core dumped) pytorch 또는 pytorch_lightning 사용할 때, gpu 사용 시에, Segmentation fault (core dumped) 오류가 뜰 때가 있다. 원인은 torch version과 torchvision version이 호환되지 않아서 이고 이 경우에는 torch version을 먼저 확인해주는 것이 좋다. 위와 같이 torch version이 cuda 또는 gpu를 호환하는 게 확인됐다면 torchvision 에서 같은 version의 cuda를 호환하지 않을 가능성이 있다. (만약, torch version이 cuda를 호환하지 않는다면 torch를 재설치해야 한다.) 그렇다면 기존의 torchvision을 pip uninstall torchvision 명령어를 통해 삭제하고 아래 .. 2022. 11. 1. torch.cuda.is_available()이 False일 때 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의 코드를 프롬프트에 복사,.. 2022. 2. 22. 이전 1 다음 반응형
