ABOUT ME

-

Today
-
Yesterday
-
Total
-
  • [에러] Pytorch 설치 시 PackagesNotFoundError
    개발 2022. 10. 4. 21:53

     

    PackagesNotFoundError

    PackagesNotFoundError는 conda install에서 가장 흔히 발생하는 오류 중 하나로 conda에서 패키지를 다운로드하려는 기본 채널에 패키지가 존재하지 않는 경우 발생한다. 나의 경우 pytorch 설치에서 에러가 발생했다.

     

     

    해결방법

    1. c 채널 옵션에 conda-forge를 주어서 패키지를 다운로드한다. conda-forge 는 anaconda에서 쉽게 설치할 수 있도록 검증된 파이썬 패키지들을 모아 놓은 하나의 채널이다.

    conda install -c conda-forge 패키지명

     

    2. 위 방법으로 파이토치를 설치하려고 했으나 에러가 나서 아래 사이트를 참고하였다.

    https://pytorch.org/get-started/locally/

     

    자신의 환경에 맞게 체크를 하면 그에 맞는 명령어를 추천해준다. GPU가 없는 경우 CUDA가 아닌 CPU를 선택한다. 추천받은 명령어를 수정하여 내가 원하는 버전으로 pytorch를 설치한다.

    conda install pytorch=1.7.1 cudatoolkit=11.6 -c pytorch -c conda-forge
     

    그러나 이 방법으로 설치한 pytorch에서는 자꾸 cuda 사용이 불가하다고 떴다. 아마 내가 원하는 pytorch 1.7.1 버전과 cudatoolkit 버전이 잘 안맞아서 였던 것 같기도 하다.

     

    3. 따라서 내가 원하는 pytorch 버전을 설치하기 위해 어떤 명령어를 써야하는지를 알아보기 위해 아래 사이트를 참고하였다.

    https://pytorch.org/get-started/previous-versions/

     

    conda install pytorch==1.7.1 cudatoolkit=11.0 -c pytorch
     

    나는 torchvision, torchaudio 패키지는 필요 없기 때문에 이를 지우고 pytorch와 cudatoolkit만 설치하고, cuda 사용이 가능한지 확인해보았더니 cuda가 잘 불러와지는 것을 확인할 수 있다.

     

     

    참고로 CUDA는 GPU 사용 여부와 관계가 있는데 NVIDIA의 그래픽 카드가 있어야 사용할 수 있다. 윈도우에서 [장치 관리자] - [디스플레이 어댑터]로 들어가면 그래픽 카드를 확인할 수 있다.

Designed by Tistory.