Linux에서 PyTorch를 설치하는 방법

1. PyTorch 소개

PyTorch는 Torch를 기반으로 하는 오픈 소스 Python 기계 학습 라이브러리로, 자연어 처리 및 기타 애플리케이션에 사용됩니다. 2017년 1월, FAIR(Facebook Artificial Intelligence Research Institute)에서는 Torch를 기반으로 PyTorch를 출시했습니다. PyTorch의 전신은 Torch입니다. 기본 레이어는 Torch 프레임워크와 동일하지만 많은 콘텐츠가 Python으로 다시 작성되었습니다. 이는 더 유연할 뿐만 아니라 동적 그래픽을 지원할 뿐만 아니라 Python 인터페이스도 제공합니다. Torch7 팀이 개발한 이 프레임워크는 강력한 GPU 가속을 활성화할 뿐만 아니라 동적 신경망도 지원하는 Python 최초의 딥 러닝 프레임워크입니다. PyTorch는 GPU 지원으로 Numpy라고 할 수 있고, 자동 파생 기능을 갖춘 강력한 심층 신경망이라고도 볼 수 있습니다. Facebook 외에도 Twitter, CMU, Salesforce 등의 기관에서 채택했습니다.

2. 설치 단계

1. 운영 체제 선택

   PyTorch 공식 웹사이트를 확인하면 PyTorch가 Linux, Mac, Windows 플랫폼, conda, pip, 소스 코드 및 기타 설치 방법을 지원하는 것을 볼 수 있습니다. CPU, cuda, ROCm 컴퓨팅 플랫폼도 지원합니다. 환경 선택을 클릭하면 현재 Linux 시스템만 모든 언어, 모든 설치 방법 및 모든 컴퓨팅 플랫폼을 지원한다는 것을 알 수 있으므로 Linux 운영 체제를 선택했습니다. 시스템 환경. 또한 기계 학습 계산에는 상위 버전의 glibc가 필요하고, centos의 커널 및 glbic 버전은 더 낮으며, Ubuntu의 커널 버전은 최신이므로 기계 학습 호스트에는 Ubuntu 운영 체제를 사용하는 것이 좋습니다. 현재 cuda 업데이트에서 지원하는 최소 Ubuntu 버전은 18.04이므로 Ubuntu 18.04 이상의 운영체제 사용을 권장합니다.

wuhs@s169:~$ cat /etc/os-release
NAME="Ubuntu"
VERSION="18.04.6 LTS (Bionic Beaver)"

2, Anaconda3 설치

 &emsp ; 위에서 본 것처럼 PyTorch는 다양한 설치 방법을 지원합니다. 블로거는 Anaconda3를 먼저 설치하는 것을 권장합니다. 필요에 따라 다양한 PyTorch 버전을 설치하는 것이 좋습니다. .가상 환경은 상호 영향을 미치지 않습니다. Ubuntu 환경에서 anaconda를 설치하려면 블로그 게시물 Ubuntu Anaconda3 Installation을 참조하세요.

wuhs@s169:~$ wget https://mirrors.bfsu.edu.cn/anaconda/archive/Anaconda3-2022.10-Linux-x86_64.sh
wuhs@s169:~$ sh Anaconda3-2022.10-Linux-x86_64 .sh
wuhs@s169:~$ source ~/.bashrc

3. Python 버전 확인

  PyTorch 버전마다 Python 버전이 다르기 때문에 anaconda3 설치 후 현재 버전 Python 버전을 확인합니다. 기본 초기화는 anaconda3의 현재 릴리스에 해당하는 최신 버전의 Python입니다. 물론 conda를 사용하여 필요한 Python 환경 버전을 생성할 수도 있습니다. 우리는 torchvision에서 PyTorch, torchvision, Python 버전 일치 요구 사항을 확인합니다.

(base) wuhs@s169:~$ python -V
Python 3.9.13

4. PyTorch를 설치합니다

  두 번째 단계는 PyTorch 공식 웹사이트에서 다음과 같습니다. 운영 체제를 선택할 수 있습니다. 설치 방법, 프로그래밍 언어 및 컴퓨팅 플랫폼을 결정한 후 해당 설치 명령이 생성됩니다.

(base) wuhs@s169:~$ conda install pytorch torchvision torchaudio cpuonly -c pytorch
패키지 메타데이터 수집(current_repodata.json): done
해결 환경: done

## 패키지 계획 ##

진행([y]/n) y
…

5. 버전 확인

(base) wuhs@s169:~$ python
Python 3.9.13(메인, 2022년 8월 25일, 23:26) 10)
[GCC 11.2.0] :: Anaconda, Inc. on linux
자세한 내용을 보려면 "help", "copyright", "credits" 또는 "license"를 입력하세요.
>>> import torch
> >>torch.version
‘1.13.1’
>>

3. 특정 버전 설치

1. 가상 환경 만들기

(기본) wuhs@s169: ~$ conda create -n pytorch python=3.9
…
(base) wuhs@s169:~$ conda activate pytorch
(pytorch) wuhs@s169:~$

2 지정된 버전의 PyTorch

&emsp를 설치합니다. ;  특정 버전의 PyTorch를 설치할 때 GitHub 공식 웹사이트의 PyTorch 채널에서 일치하는 버전을 확인해야 합니다. Conda를 설치할 때 버전 번호를 지정하면 anaconda 공식 웹사이트에서 확인할 수 있습니다. . PyTorch 버전과 TorchAudio 간의 해당 관계는
TorchAudio를 참조하세요. 물론 잘못된 버전을 지정하면 설치 시 오류가 발생하는데, 오류 메시지를 통해 어떤 소프트웨어 버전이 잘못된지 확인한 뒤 공식 홈페이지에 가서 수정 사항을 확인하고 재설치할 수 있다.

(pytorch) wuhs@s169:~$ conda install pytorch2.12.0 torchvision=0.13.0 torchaudio0.12.0 cpuonly -c pytorch

위 내용은 Linux에서 PyTorch를 설치하는 방법의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!