코딩을 하다가 'pytorch-lightning' 이라는 라이브러리를 사용해보려고 설치했는데 무슨 "Segmentation fault (core dumped)" 라는 에러가 뜨기도 하고 "Import error ~ undefined symbol ~~" 과 같은 에러도 발생해서 StackOverflow,Github, 등등을 돌아다니던 중 유용한 코드가 있길래 가져왔습니다!
$python -m torch.utils.collect_env위 커맨드를 입력하면 Pytorch의 버전은 물론 CUDA 버전 및 torch.cuda.is_available() 의 정보도 알려주며 torch와 관련된 라이브러리의 버전 정보까지(중요) 알려줍니다.
상당히 종합적인 정보를 볼 수 있어서 버전 충돌과 같은 이슈를 해결하는데 어느정도 도움이 되지 않을까 싶습니다.
일반적으로는 아래와 같은 커맨드를 이용하여 확인할 수 있지만, torch와 관련된 라이브러리 버전만 보기엔 정말 좋은거 같습니다.
$pip freeze
$pip list마지막으로 제 컴퓨터에 설치되어 있는 pytorch 버전 및 기타 정보들은 아래와 같이 나오네요
발견한 곳 : https://discuss.pytorch.org/t/segmentation-fault-core-dump-when-using-model-cuda/122049
'파이썬(Python) > 파이썬 관련' 카테고리의 다른 글
| [Hugging Face🤗] Padding & Truncation 정리 (0) | 2023.06.28 |
|---|---|
| GPU 메모리 확보하기 (0) | 2023.06.13 |
| Selenium으로 웹사이트(디시인사이드) 크롤링하기 (16) | 2023.04.15 |
| [Python] 가상환경(Virtual Environment) 세팅하기 (0) | 2021.08.04 |
