Local pc install
1. CONDA 설치 : 설치환경에 따라 충돌일 발생할 수 있어 가상환경을 구축할 수 있는 CONDA 설치
2. Visual studio 설치
3. CUDA 설치
4. cuDNN 설치
5. 토치 설치 : 1.8.1이 CUDA와 가장 충돌이 없다고 함.
https://pytorch.org/get-started/previous-versions/
v1.8.1
Conda
OSX
# conda
conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 -c pytorch
Linux and Windows
# CUDA 10.2
conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 cudatoolkit=10.2 -c pytorch
# CUDA 11.3
conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 cudatoolkit=11.3 -c pytorch -c conda-forge
# CPU Only
conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 cpuonly -c pytorch
* 토치 설치 확인
https://chancoding.tistory.com/90
윈도우에 PyTorch 설치, GPU 설정, 자세하게
윈도우에 파이토치를 설치하고 GPU를 사용하는 방법에 대해서 알아보겠습니다. 1. 가상환경 만들기 (생략 가능) 기본적으로 저는 가상 환경을 나누어서 작업을 진행하는 것을 선호합니다. 그래서
chancoding.tistory.com