Bigdata

AMD AI 9 관련 (370 등) CPU ROCm 지원, 설치 방법

올엠 2025. 12. 19. 14:30
반응형

이 과정은 Stability Matrix가 관리하는 ComfyUI 가상 환경의 기존 PyTorch를 제거하고, AMD ROCm 7.1.1을 지원하는 전용 버전으로 교체하는 작업이다.

AMD ROCm 7.1.1 버전은 기존에 지원하지 않던 모델을 다수 지원해주기 시작하면서 UMPC 뿐만이 아니라, 가성비 노트북등에서도 AI 성능을 보다 끌어 올릴 수 있게 된다.

📋 1. 사전 준비 (Prerequisites)

설치를 시작하기 전에 다음 두 가지가 반드시 준비되어야 합니다.

  2. Python 3.12 환경:
    • ROCm 7.1.1은 Python 3.12를 요구합니다.
    • 설정: Stability Matrix에서 패키지를 설치하거나 설정할 때, 사용하는 Python 버전이 3.12로 설정되어 있는지 확인하세요.

🛠️ 2. 설치 단계 (Installation Steps)

모든 작업은 Stability Matrix가 설치한 ComfyUI 패키지 폴더 내에서 cmd (명령 프롬프트)를 열고 진행해야 합니다.

1단계: 가상 환경 활성화

ComfyUI 폴더에서 터미널을 열고 다음 명령어를 입력해 가상 환경(venv)으로 진입합니다.

source .venv\Scripts\activate

2단계: 기존 PyTorch 삭제

충돌을 방지하기 위해 기존에 설치된 PyTorch 관련 패키지를 모두 삭제합니다.

 
pip uninstall -y torch torchaudio torchvision

3단계: ROCm SDK 컴포넌트 설치

AMD Radeon 리포지토리에서 ROCm SDK 핵심 라이브러리들을 설치합니다.
(※ 이 과정에서 의존성 에러(dependency resolver error)가 발생할 수 있으나, 설치 완료 메시지(Successfully installed)가 뜬다면 무시해도 됩니다.)

 
pip install --no-cache-dir https://repo.radeon.com/rocm/windows/rocm-rel-7.1.1/rocm_sdk_core-0.1.dev0-py3-none-win_amd64.whl 
pip install --no-cache-dir https://repo.radeon.com/rocm/windows/rocm-rel-7.1.1/rocm_sdk_devel-0.1.dev0-py3-none-win_amd64.whl pip install --no-cache-dir https://repo.radeon.com/rocm/windows/rocm-rel-7.1.1/rocm_sdk_libraries_custom-0.1.dev0-py3-none-win_amd64.whl
pip install --no-cache-dir https://repo.radeon.com/rocm/windows/rocm-rel-7.1.1/rocm-0.1.dev0.tar.gz
 

4단계: ROCm 대응 PyTorch 설치

ROCm 7.1.1과 호환되는 PyTorch, TorchAudio, TorchVision을 설치합니다.

 
pip install --no-cache-dir https://repo.radeon.com/rocm/windows/rocm-rel-7.1.1/torch-2.9.0+rocmsdk20251116-cp312-cp312-win_amd64.whl
pip install --no-cache-dir https://repo.radeon.com/rocm/windows/rocm-rel-7.1.1/torchaudio-2.9.0+rocmsdk20251116-cp312-cp312-win_amd64.whl
pip install --no-cache-dir https://repo.radeon.com/rocm/windows/rocm-rel-7.1.1/torchvision-0.24.0+rocmsdk20251116-cp312-cp312-win_amd64.whl
 

✅동작 확인 (Verification)

설치가 끝났으면 Python을 실행해 GPU가 정상적으로 인식되는지 확인합니다.

  1. Python 실행:
     
    python
  2. 확인 코드 입력: 
    import torch 
print(f"Torch Version: {torch.__version__}") 
print(f"Is HIP available: {torch.cuda.is_available()}") 
print(f"Device Name: {torch.cuda.get_device_name(0) if torch.cuda.is_available() else 'N/A'}")

정상 출력 예시:

  • Torch Version: 2.9.0+rocmsdk20251116
  • Is HIP available: True (이 부분이 True면 성공~~)
  • Device Name: 사용 중인 그래픽카드 이름 (예: AMD Radeon RX 9060 XT)

🚀 ComfyUI 실행

Stability Matrix에서 ComfyUI를 실행한 후 콘솔 로그에서 아래와 같은 로그가 보이면 성공이다.

  • AMD arch: gfx1200 (사용자 아키텍처에 따라 다름)
  • ROCm version: (7, 1)
  • Device: cuda:0 AMD Radeon ... : native

⚠️ 주의사항
실행할 때마다 'K-Sampler'를 매번 다시 컴파일하는 등 동작이 다소 불안정할 수 있다. 실사용 목적보다는 테스트 목적으로 시도해보기 바란다.

반응형