[PyTorch/Python] conda list의 torch 버전과 pip show torch 버전이 다를 때

[PyTorch/Python] conda list의 torch 버전과 pip show torch 버전이 다를 때

 

pip show torch에 출력된 파이토치 버전과 conda list로 출력된 파이토치 버전이 다를 때 해결법이다.

 

1. 문제 상황

파이토치를 설치했고, conda list 로도 출력이 되는데 no module name 'torch' 에러가 자꾸 떴다.

 

1.1 pip torch 버전 확인

pip show torch

위의 명령어로 torch 버전을 확인해보니 아래처럼 나왔다.

Name: torch
Version: 2.2.1
Summary: Tensors and Dynamic neural networks in Python with strong GPU acceleration
Home-page: https://pytorch.org/
Author: PyTorch Team
Author-email: packages@pytorch.org
License: BSD-3
Location: /home/user/.local/lib/python3.10/site-packages
Requires: filelock, fsspec, jinja2, networkx, nvidia-cublas-cu12, nvidia-cuda-cupti-cu12, nvidia-cuda-nvrtc-cu12, nvidia-cuda-runtime-cu12, nvidia-cudnn-cu12, nvidia-cufft-cu12, nvidia-curand-cu12, nvidia-cusolver-cu12, nvidia-cusparse-cu12, nvidia-nccl-cu12, nvidia-nvtx-cu12, sympy, triton, typing-extensions
Required-by: timm, torchaudio, torchvision

 

1.2 conda list로 conda torch 버전 확인

conda list

위의 명령어로 torch 버전을 확인해보니 아래처럼 나왔다.

python                    3.10.16              he870216_1  
pytorch                   1.11.0          py3.10_cuda11.3_cudnn8.2.0_0    pytorch

 

2. 문제 원인

필자는 pytorch 1.11.0 버전이 필요해서 공식 페이지에 나와있는 명령어로 해당 버전의 pytorch를 설치했었다.

그러나 pip 에서는 다른 토치 버전이 설치되어있었다.

pytorch 설치 후 다른 패키지를 설치할 때 버전 충돌이 있어 내부적으로 어떤 작용이 있었을 것이라 추측된다.

정확하게는 모르겠지만 이런 저런 패키지를 설치하다가 버전 충돌이 일어난 듯 하다.

 

3. 해결 방법

해결 방법은 pip 단에서 torch를 지우는 것이다.

 

3.1. pip uninstall torch

pip uninstall torch

위 명령어를 실행하여 pip 단에 있는 torch를 삭제한다.

 

3.2. 패키지 완전 삭제

rm -rf ~/.local/lib/python3.10/site-packages/torch*

위 명령어로 완전히 삭제한다.

 

4. 해결 완료 

이제 pip show torch를 다시 실행해보자.

Name: torch
Version: 1.11.0
Summary: Tensors and Dynamic neural networks in Python with strong GPU acceleration
Home-page: https://pytorch.org/
Author: PyTorch Team
Author-email: packages@pytorch.org
License: BSD-3
Location: /home/user/anaconda3/envs/sama/lib/python3.10/site-packages
Requires: typing_extensions
Required-by: timm, torchaudio, torchvision

 

원하는 버전이 출력되는 것을 볼 수 있다.