치즈의 AI 녹이기

내 환경에 맞는 pytorch 버젼 깔기 본문

인공지능 대학원생의 생활/구글링

내 환경에 맞는 pytorch 버젼 깔기

개발자 치즈 2022. 7. 14. 22:03
pip install torch --upgrade version==1.10.1

1. 내 CUDA 버젼 확인하기

watch nvidia-smi

결과 화면 및 버젼 확인

2. 해당 사이트에서 자신의 CUDA버젼과 호환이 되는 torch 버젼 알기

cu111은 CUDA 버젼 11.1의 경우 이므로, 10.1의 경우 101로 교체하면 됨. 

(https://download.pytorch.org/whl/cu111/torch_stable.html)

나는 CUDA 11.0으로 검색해서 맞는 명령어를 찾아준다. 

 

3. 내 conda 가상환경에 깔린 패키지 목록 확인하여 torch 버젼 확인 

conda env list

현재 ipython으로 import torch해서 버젼 확인했을 때는 1.7.0인데

가상환경에 깔린 torch는 1.9.0이 있는 요상한 상태.

 

4. 원하는 버젼으로 다시 torch 모듈 버젼 업데이트 하기 

일단 이미 설치되어 있는 pytorch, torchvision, cudatookit을 다 unintall 해주고

pip uninstall pytorch torchvision cudatookit

아까 pytorch 홈페이지에서 찾아놨던 명령어를 실행해준다!

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