AI

Pytorch 설치하기

leonhong 2022. 11. 21. 19:38

현재 사용중인 버전의 pytorch 설치하기

 

python : 3.8

pytorch : 1.9.1

cuda : 10.2

 

anaconda 명령어

pytorch 설치

conda create --name pytorch python=3.8

conda activate pytorch

conda install pytorch==1.9.1 torchvision==0.10.1 torchaudio==0.9.1 cudatoolkit=10.2 -c pytorch

 

보조툴 설치 (scikit-learn, Pandas, Matplotlib)

conda install -c conda-forge scikit-learn=1.1

conda install -c anaconda pandas=1.4

conda install -c conda-forge matplotlib=3.5

 

하위호환을 위한 라이브러리 설치

conda install scipy=1.6.1