[torchvision] RegNet, EfficientNet, FX Feature Extraction 등 설치 안될 때(torchvision 새 버전 install하기)

참고

Releases · pytorch/vision

 

efficientnet과 regnet이 버전의 문제로 설치가 안됐었다. 0.11.0 버전부터 가능한 모델이었다.

그래서 위의 링크를 참고하여 아래의 명령어를 실행하였지만 오류가 발생했다.

 

pip install torchvision==0.11.0

오류 내용:

WARNING: The candidate selected for download or install is a yanked version: 'torchvision' candidate (version 0.11.0 at https://files.pythonhosted.org/packages/8d/26/78230f895a1a6cfba22cbf088abf5d3ab5ef0fe29ea0a6a741ec3a633a2e/torchvision-0.11.0-cp38-cp38-manylinux1_x86_64.whl#sha256=de8b520f4ae872e8399aa7f4885c8998a8dd2b7d35bcf48ecbc700f05ef0261a (from https://pypi.org/simple/torchvision/))
Reason for being yanked: Dependency issue, depends on a version of torch that does not exist on pypi
Collecting torchvision==0.11.0
  Using cached torchvision-0.11.0-cp38-cp38-manylinux1_x86_64.whl (23.3 MB)
Requirement already satisfied: numpy in /opt/conda/lib/python3.8/site-packages (from torchvision==0.11.0) (1.20.0)
Requirement already satisfied: pillow!=8.3.0,>=5.3.0 in /opt/conda/lib/python3.8/site-packages (from torchvision==0.11.0) (8.3.2)
ERROR: Could not find a version that satisfies the requirement torch==1.10.0+cu102 (from torchvision==0.11.0) (from versions: 1.4.0, 1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1, 1.8.0, 1.8.1, 1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2)
ERROR: No matching distribution found for torch==1.10.0+cu102 (from torchvision==0.11.0)
Note: you may need to restart the kernel to use updated packages.

그래서 코랩에서도 이 모델들이 임포트되지 않는지 확인하였으나 코랩에서는 정상적으로 임포트가 가능했다.

따라서 코랩에서의 torchvision 버전을 확인해보니 0.11.1이었다.

 

혹시나해서 현재 작업중이던 주피터랩에서도 0.11.1버전으로 설치해보니 잘 설치가 되었다.

 

결론

+) 아래 과정을 이행하기 전 torch 버전을 먼저 올려주어야 한다!

torch와 torchvision은 서로 dependence하기 때문에 서로 호환되는 버전을 설치해주어야 한다.

맞지 않는 버전이 설치되어 있으면 충돌이 일어나 제대로 설치되지 않는다.

 

0. 따라서 torch 1.10.0 버전을 설치한다.

pip install torch==1.10.0

 

1. 아래 코드를 실행한다.

pip install torchvision==0.11.1

2. 커널을 재시작(restart)한다.

3. 다시 torchvision을 install할 필요없이, 그냥 모델을 바로 임포트해서 사용한다.