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 명령어를 통해 삭제하고
아래 사이트에서
cuda=11.6 위치의 버젼을 조정해주거나
이전 버젼의 torch를 사용한다면
위 사이트에서 맞는 운영체제와 cuda 버젼에 맞는 라이브러리를 찾아 설치해주면 된다.
만약 위의 1.9.0+cu102에 맞는 torchvision 을 설치하고자 한다면
노란색으로 표시한 부분의 명령어를 서버나 터미널에 입력해주면 된다.
728x90
'Data > Errors' 카테고리의 다른 글
[Google Colab] error: invalid zip file with overlapped components (possible zip bomb) (0) | 2023.10.02 |
---|---|
ImportError: No module named sqlalchemy (0) | 2023.09.26 |
[Vs Code] conda 가상 환경 변경 안 될 때 (0) | 2022.09.26 |
Vue 머스태시(mustache) 랜더링 안될 때 (0) | 2022.07.19 |
import cv2 안될 때 (0) | 2022.04.05 |