windows11 에서 ROCm 사용해서 pytorch

Windows 11에서 ROCm 사용하여 PyTorch 설치 방법

참고: Windows 11에서 ROCm을 사용하여 PyTorch를 설치하는 것은 공식적으로 지원되지 않습니다.

하지만 다음 단계를 통해 설치할 수 있습니다.

1. ROCm 설치:

• AMD ROCm 공식 웹사이트에서 최신 버전의 ROCm을 다운로드합니다

• 다운로드한 설치 파일을 실행하여 ROCm을 설치합니다.

• 설치 과정에서 CUDA 지원을 선택적으로 설치할 수 있습니다.

2. Python 설치:

• Python 공식 웹사이트에서 최신 버전의 Python을 다운로드합니다.

• 다운로드한 설치 파일을 실행하여 Python을 설치합니다.

3. PyTorch 설치:

• PyTorch 공식 웹사이트에서 Windows 11 + ROCm + Python 버전에 맞는 PyTorch 바이너리 패키지를 다운로드합니다.

• 다운로드한 바이너리 패키지를 압축 해제합니다.

• 명령 프롬프트 또는 터미널을 실행하고 압축 해제된 폴더로 이동합니다.

• 다음 명령을 실행하여 PyTorch를 설치합니다.

pip install torch==<pytorch_version>-rocm

참고:

• <pytorch_version>은 다운로드한 PyTorch 버전으로 변경해야 합니다.

• pip 명령이 없으면 Python 설치 과정에서 함께 설치되었는지 확인해야 합니다.

• pip 명령을 사용하여 다른 PyTorch 패키지도 설치할 수 있습니다.

4. 환경 설정:

• PyTorch를 ROCm과 함께 사용하려면 환경 변수를 설정해야 합니다.

• 다음 명령을 사용하여 ROCm 라이브러리 경로를 설정합니다.

set PATH=C:\Program Files\AMD\ROCm\bin;%PATH%

참고:

• C:\Program Files\AMD\ROCm\bin은 ROCm 설치 경로에 따라 변경해야 합니다.

• 영구적으로 환경 변수를 설정하려면 시스템 환경 변수에 ROCm 라이브러리 경로를 추가해야 합니다.

5. 테스트:

• 다음 코드를 실행하여 PyTorch와 ROCm이 제대로 설치되었는지 확인합니다.

Python

import torch print(torch.cuda.is_available())

출력:

True

참고:

• True가 출력되면 PyTorch와 ROCm이 제대로 설치되었음을 의미합니다.

• False가 출력되면 설치 과정을 다시 확인해야 합니다.

문제 해결:

• 설치 과정에서 문제가 발생하면 ROCm 및 PyTorch 공식 문서를 참고하거나 관련 커뮤니티에 질문을 게시하세요.

• ​