[기타]docker로 딥러닝 환경 구축하기(4) pytorch 등 라이브러리 설치

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

PyTorch

 

모든 cuda 버전과 pytorch가 잘 호환되면 괜찮겠지만 이게 그렇진 않은거 같습니다 

공식 홈페이지에서 본인이 설치한 cuda 버전을 확인하시고 호환되는 버전을 찾아주시면 됩니다

 

 

conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch

왠지 11.4는 없어서 11.3으로 설치해주겠습니다. 결론은 다행히도 큰 문제는 안생기네요 

 

설치가 완료되었으면 cuda가 torch랑 잘 연결되었는지 확인을 해야겠습니다 

 

ipython
import torch
torch.cuda.is_available()
>>>True

True가 나오시면 끝