간혹 타인의 소스코드를 구현하면서 버전이 호환되지 않으면서 동작하지 않는 경우가 종종 발생한다.
되도록 구현한 사람과 동일한 환경을 먼저 설정한 뒤 소스코드를 분석하는 것이 시간낭비 하지않는 방법이라 생각한다.
한가지 주의할 점은 Python과 Pytorch 간 호환되는 버전인지도 확인한 뒤 설치하여야 한다.
Python 설치방법
conda install python="Python version" → ex) conda install python=3.7
Pytorch 설치방법
<pip3>
pip3 install http://download.pytorch.org/whl
위의 링크를 먼저 적어준 뒤,
/cpu/torch-0.4.1-cp36-cp36m-win_amd64.whl
위의 예제처럼 원하는 링크를 추가해주며 된다.
→ ex) pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp36-cp36m-win_amd64.whl
<Anaconda>
conda install pytorch='원하는버전' → ex) conda install pytorch=0.4.1 -c pytorch
torchvision 설치방법
<Anaconda>
conda install torchvision -c pytorch
<pip>
pip install torchvision
Pytorch 버전별 설치방법 참고
'개인공부 > Python' 카테고리의 다른 글
im2col 함수 (0) | 2022.01.18 |
---|---|
Pytorch - 텐서 조작하기 (0) | 2022.01.18 |
RuntimeError (CUDA) (0) | 2022.01.13 |
파라미터 저장을 위한 예제 (hdf5) (0) | 2021.12.20 |
(1) 변수와 리스트 (0) | 2021.12.14 |