AI

아나콘다 주피터 pytorch torchvision 설치 오류

초이짬 2021. 3. 4. 16:44
728x90

아나콘다로 주피터 띄우고 jupyter 상에서 !pip install torch 로

 

파이토치 설치 하고 같은 방식으로 torchvision 설치 하면 버전 문제 에러 난다

 

이런거.... torch버전은 1.7.1+cpu

ERROR: Could not find a version that satisfies the requirement torch==1.4.0 (from torchvision) (from versions: 0.1.2, 0.1.2.post1, 0.1.2.post2, 1.7.1) ERROR: No matching distribution found for torch==1.4.0 (from torchvision)

 

윈도우 기준으로 아나콘다 프롬프트 띄우고 

 

conda install pytorch torchvision -c pytorch

 

이렇게 설치하면 1.6 기가 정도 설치 할려고 하고 y 누르면 되는데

 

다 끝나갈떄 쯤에 c드라이브를 베이스로 잡아서 그런지 쓰기 권한 오류 떨어진다

 

그럴땐 현재 떠있는 윈도우 쉘을 관리자 권한으로 다시 띄우고 하면 설치 된다......

 

짝이 맞는 버전이 torch=1.7.1, torchvision=0.8.2

 

    package                    |            build
    ---------------------------|-----------------
    conda-4.9.2                |   py37haa95532_0         2.9 MB
    cudatoolkit-11.0.221       |       h74a9793_0       627.0 MB
    libuv-1.40.0               |       he774522_0         255 KB
    ninja-1.10.2               |   py37h6d14046_0         246 KB
    pytorch-1.7.1              |py3.7_cuda110_cudnn8_0      1007.2 MB  pytorch
    torchvision-0.8.2          |       py37_cu110         7.2 MB  pytorch
    typing_extensions-3.7.4.3  |     pyha847dfd_0          25 KB
    ------------------------------------------------------------
                                           Total:        1.61 GB

 

요렇게 짝 맞춰 준다

728x90

'AI' 카테고리의 다른 글

OpenCV 설치 에러 import cv2  (0) 2020.11.23