Pytorch 설치 (Solving Environment 멈춤 현상)

문제 상황

conda 환경에서 파이토치 설치 중 solving environment에서 작업이 무한 지속되는 문제가 발생했다. 검색해보니 관련 문제가 빈번하게 나타나는 듯 했다.

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch

트러블 슈팅 참조 링크

위를 참고해보면 기존 설치 커맨드에 '-c conda-forge' 문구를 추가하여 해결하고 있다. conda-forge 채널은 커뮤니티에서 유지 보수하는 conda 패키지 저장소이다. 기본 제공되는 defaults보다 더 많은 패키지를 포함한다.

아무튼 위 커맨드를 통해서 성공적으로 설치가 되었다. 아마 기존에는 디폴트 파이토치 채널에서 설치 패키지 버전을 가져오는데 있어서 문제가 발생했던 것 같다. CUDA 관련 패키지 (cudatoolkit=11.0)가 디폴트 채널에 없는 경우에는 conda-forge에서 가져오게끔함으로써 버전 충돌을 방지한 것으로 보인다.

결론

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch -c conda-forge

기존 커맨드에 마지막에 '-c conda-forge'를 추가한다. 참고로 conda list pytorch 커맨드 입력 시 파이토치 버전이 출력된다면 정상적으로 설치가 된 것이다.