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

'AI' 카테고리의 다른 글

Python에서 Opencv 설치하기  (0) 2022.12.08
Numpy Index Slicing을 이용한 빠른 접근  (0) 2022.07.28
pytorch 1.9.1 cu102 버전 설치 방법  (0) 2021.11.04