Centos의 Pytorch 구성 단계

CentOS 시스템에 Pytorch를 배포하는이 기사는 자세한 단계를 제공합니다.

1. 아나콘다 설치

1. Centos 용 Anaconda 설치 패키지를 다운로드하려면 Anaconda 공식 웹 사이트 를 방문하십시오.
2. 설치 프로그램을 실행하고 프롬프트를 따라 설치를 완료하십시오. 후속 환경 변수 구성을 용이하게하기 위해 "Just Me"설치 방법을 선택하는 것이 좋습니다.

2. 가상 환경 생성

1. 터미널을 열고 다음 명령을 사용하여 pytorch_env 라는 가상 환경을 만들고 Python 버전 (예 : Python 3.10)을 지정하십시오.

 콘다 생성 -n pytorch_env python = 3.10

2. 가상 환경 활성화 :

 콘다 활성화 pytorch_env

3. 미러 소스 구성 (선택 사항)

다운로드 속도를 높이려면 국내 미러 소스를 구성하는 것이 좋습니다. .condarc 파일 (일반적으로 사용자의 홈 디렉토리에 위치)을 편집하고 다음을 추가하십시오.

 <code>channels: - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forges/ show_channel_urls: true auto_activate_base: false</code>

4. Pytorch 설치

1. CUDA 버전 확인 : 다음 명령을 사용하여 CUDA 버전을보십시오.

 NVIDIA-SMI

CUDA가 설치되지 않은 경우 요구 사항에 따라 설치하십시오.

2. Pytorch 다운로드 및 설치 :

   • Pytorch 공식 웹 사이트 또는 Tsinghua University 오픈 소스 소프트웨어 미러링 사이트를 방문하여 필요한 Pytorch 버전을 얻으십시오.
   • CUDA 버전과 일치하는 Pytorch 버전을 선택하십시오. 예를 들어, Cuda 12.1은 Pytorch의 Cuda 12.1 버전에 해당합니다.

   pip 사용하여 설치합니다 (예로 Cuda 12.1 복용) :

 PIP Torch TorchVision Torchaudio -Index-url https://download.pytorch.org/whl/cu121

또는 conda 사용하여 설치 :

 콘다 설치 Pytorch Torchvision Torchaudio Cudatoolkit = 12.1 -C Pytorch

3. 설치 확인 : 가상 환경을 활성화 한 후 Python 대화식 환경을 입력하고 다음 명령을 실행하여 설치를 확인하십시오.

 토치 수입
인쇄 (토치 .__ 버전 __)
print (torch.cuda.is_available ())

torch.cuda.is_available() Pytorch가 성공적으로 설치되어 GPU를 사용할 수 있음을 나타 내기 위해 True 반환합니다.

5. 문제 조사

• Numpy Torch 버전과의 충돌 : Numpy 및 Torch 버전이 호환되지 않으면 Numpy를 다운 그레이드하십시오.

 PIP 설치 "Numpy "

• nvcc 명령을 찾을 수 없음 : nvcc 명령은 CUDA 설치 오류 또는 환경 변수를 올바르게 구성하지 않았기 때문일 수 있습니다. CUDA 설치 경로가 시스템 환경 변수에 추가되었는지 확인하십시오.

설치 과정에서 발생하는 다른 문제를 해결하려면 공식 Pytorch 문서 또는 커뮤니티 포럼을 참조하십시오. 위의 단계가 Centos 시스템에서 Pytorch 환경을 부드럽게 구성하는 데 도움이되기를 바랍니다.

위 내용은 Centos의 Pytorch 구성 단계의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!