Ubuntu Pytorch Install, Uninstall (python 3.8)

#2024-05-21

 

Pytorch Install

• Ubuntu 20.04 LTS

• CUDA 11.8
• cuDNN 8.9.7
• python 3.8
• torch v2.0.0+cu118

 

pytorch 2.0.0 상위의 버전을 받을 경우, 아래와 같은 오류를 접할 수 있음

ERROR: Package 'networkx' requires a different Python: 3.8.10 not in '>=3.9'

 

따라서 cuda 11.8을 호환하고, networkx package의 error를 벗어날 수 있는 2.0.0 버전을 설치

install previous version of pytorch를 찾아 들어가서 찾아보면 사진과 같이 설치 명령어들을 볼 수 있음

 

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

Previous PyTorch Versions

Pytorch Install

 

해당하는 명령어를 찾아서 설치

python3에 설치해 주기 위해서 pip → pip3로 변경 (pip3의 version이 python3인 경우)

pip version이 설치하고자 하는 python3.8이라면 pip를 사용해도 상관없음

pip3 install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118

 

제대로 설치됐는지 확인

torch.cuda.is_available() 시, True가 나오면 제대로 설치 완료한 것

import torch
torch.__version__
torch.cuda.is_available()

Pytorch 설치 확인

 

---

#추가

#2024-05-22

Pytorch Uninstall

설치된 torch부터 제거

pip3 uninstall torch

 

pip3 list를 통해 설치된 torch를 확인하여 마저 제거

pip3 list

pip3 uninstall torchaudio
pip3 uninstall torchvision

pip3 list에서 찾은 torchaudio, torchvision

 

 

참조

파이토치 + 쿠다 연결함 !!!!