Windows 11에서 AMD ROCm 설치하기: WSL2 환경에서 AMD GPU로 AI 작업 준비하기

Windows 11에서 AMD ROCm 설치하기: WSL2 환경에서 AMD GPU로 AI 작업 준비하기

작성일: 2025년 4월 11일

AMD GPU로 AI 작업을 하려는 Windows 사용자들을 위한 완벽 가이드입니다. 이 글에서는 Windows 11의 WSL2 환경에서 AMD ROCm을 설치하고 PyTorch와 함께 사용하는 방법을 단계별로 안내합니다.

목차

• 소개
• 시스템 요구사항
• WSL2와 Ubuntu 설치
• AMD ROCm 설치하기
• PyTorch와 ROCm 설정
• WSL 런타임 업데이트
• 설치 확인하기
• 앱 실행 시 주의사항
• FAQ
• 결론

소개

AMD ROCm(ROCm Open Compute Platform)은 AMD GPU를 활용한 고성능 컴퓨팅 및 AI 작업을 지원하는 오픈 소프트웨어 스택입니다. Windows 11에서는 WSL2(Windows Subsystem for Linux)를 통해 Linux 환경에서 ROCm을 실행할 수 있어, AI 개발자들에게 새로운 가능성을 제공합니다.

이 가이드는 윈도우 사용자가 AMD Radeon RX 7000 시리즈 GPU를 활용해 딥러닝 및 AI 작업을 수행할 수 있도록 ROCm 환경을 구축하는 방법을 상세히 설명합니다.

시스템 요구사항

시작하기 전에 다음 요구사항을 확인하세요:

요구사항 상세 정보

운영 체제	Windows 11
GPU	AMD Radeon RX 7000 시리즈 (예: Radeon RX 7900 XTX)
드라이버	AMD 소프트웨어(드라이버) 버전 24.6.1 이상
WSL2	Ubuntu 22.04 권장
기타	Windows 하이퍼바이저 플랫폼 활성화

WSL2와 Ubuntu 설치

1. WSL2 설치하기

1. 관리자 권한으로 Windows PowerShell을 열고 다음 명령어를 실행하세요:

wsl --install -d Ubuntu-22.04

1. 설치가 완료되면 사용자 이름과 비밀번호를 설정하라는 안내에 따라 진행하세요.

2. WSL2 환경 확인하기

Ubuntu가 제대로 설치되었는지 확인하려면:

wsl -l -v

이 명령어는 설치된 리눅스 배포판과 WSL 버전을 보여줍니다. Ubuntu-22.04가 VERSION 2로 표시되어야 합니다.

AMD ROCm 설치하기

WSL2 터미널에서 다음 단계를 따라 AMD ROCm을 설치하세요:

1. 패키지 업데이트 및 설치 스크립트 다운로드

sudo apt update
wget https://repo.radeon.com/amdgpu-install/6.3.4/ubuntu/jammy/amdgpu-install_6.3.60304-1_all.deb
sudo apt install ./amdgpu-install_6.3.60304-1_all.deb

2. AMD ROCm 설치

amdgpu-install -y --usecase=wsl,rocm --no-dkms

이 명령어는 WSL2 환경에 맞게 ROCm을 설치합니다. 설치 과정은 몇 분 정도 소요될 수 있으며, 오류 메시지가 없는지 확인하세요.

3. 설치 확인

rocminfo

이 명령어를 실행하면 시스템에서 인식된 AMD GPU의 정보가 표시됩니다. 성공적으로 설치되었다면 "Name: gfx1100", "Marketing Name: Radeon RX 7900 XTX"와 같은 정보가 나타납니다.

PyTorch와 ROCm 설정

AMD GPU에서 PyTorch를 사용하려면 ROCm 호환 버전을 설치해야 합니다.

1. 필수 패키지 설치

sudo apt install python3-pip -y
pip3 install --upgrade pip wheel

2. ROCm 호환 PyTorch 휠 파일 다운로드

wget https://repo.radeon.com/rocm/manylinux/rocm-rel-6.3.4/torch-2.4.0+rocm6.3.4.git7cecbf6d-cp310-cp310-linux_x86_64.whl
wget https://repo.radeon.com/rocm/manylinux/rocm-rel-6.3.4/torchvision-0.19.0+rocm6.3.4.gitf9e9a1e-cp310-cp310-linux_x86_64.whl
wget https://repo.radeon.com/rocm/manylinux/rocm-rel-6.3.4/pytorch_triton_rocm-3.0.0+rocm6.3.4.4d510c3a44-cp310-cp310-linux_x86_64.whl

참고: 최신 버전은 AMD ROCm repository에서 확인할 수 있습니다.

3. PyTorch 설치

pip3 install torch-2.4.0+rocm6.3.4.git7cecbf6d-cp310-cp310-linux_x86_64.whl torchvision-0.19.0+rocm6.3.4.gitf9e9a1e-cp310-cp310-linux_x86_64.whl pytorch_triton_rocm-3.0.0+rocm6.3.4.4d510c3a44-cp310-cp310-linux_x86_64.whl
pip3 install numpy==1.26.4

numpy 2.0과의 호환성 문제로 인해 numpy를 1.26.4 버전으로 설치합니다.

4. torchaudio 설치 (선택사항)

pip3 install torchaudio==2.4.0

WSL 런타임 업데이트

PyTorch가 AMD GPU를 제대로 인식하도록 런타임 라이브러리를 업데이트해야 합니다.

1. torch 라이브러리 위치 찾기

location=$(pip show torch | grep Location | awk -F ": " '{print $2}')
cd ${location}/torch/lib/

2. 기존 라이브러리 파일 제거 및 업데이트

rm libhsa-runtime64.so*
cp /opt/rocm/lib/libhsa-runtime64.so.1.2 libhsa-runtime64.so

주의: 실제 라이브러리 파일 이름은 /opt/rocm/lib/ 디렉토리에서 확인하세요. 버전에 따라 다를 수 있습니다.

설치 확인하기

설치가 올바르게 완료되었는지 확인하려면 다음 명령어를 실행하세요:

python3 -c 'import torch; print(torch.cuda.is_available())'

True가 출력되면 AMD GPU가 PyTorch에서 성공적으로 인식된 것입니다.

앱 실행 시 주의사항

ComfyUI와 같은 AI 앱 실행하기

ComfyUI와 같은 AI 애플리케이션을 실행할 때, requirements.txt 파일에서 torch 관련 라인을 주석 처리해야 합니다:

1. requirements.txt 파일 열기
2. 다음 라인들을 주석 처리:

   # torch==2.1.2# torchvision==0.16.2
   

3. 설치 실행:

   pip install -r requirements.txt
   

이렇게 하면 ROCm 호환 torch 버전이 덮어쓰이지 않습니다. 만약 덮어쓰였다면, 위에서 설명한 대로 ROCm 호환 torch를 다시 설치해야 합니다.

특성 AMD ROCm NVIDIA CUDA

지원 GPU	AMD Radeon RX 7000 시리즈	NVIDIA GeForce/RTX/Quadro
Windows 지원	WSL2를 통해 지원	네이티브 지원
Linux 지원	네이티브 지원	네이티브 지원
PyTorch 호환성	특수 빌드 필요	공식 지원
설치 복잡도	중간~높음	낮음~중간
커뮤니티 지원	성장 중	광범위함
오픈소스	예	부분적

FAQ

Q: Ubuntu 24.04를 사용해도 될까요?

A: 가능하지만, 일부 호환성 문제가 있을 수 있어 Ubuntu 22.04를 권장합니다.

Q: AMD Radeon RX 6000 시리즈도 지원되나요?

A: 현재 Windows WSL 환경에서는 Radeon RX 7000 시리즈가 공식 지원됩니다.

Q: Conda 환경에서도 설치할 수 있나요?

A: AMD에서 공식적으로 Conda 환경을 지원하지 않으며, GCC 12.1 설치가 추가로 필요할 수 있습니다.

Q: ROCm을 제거하려면 어떻게 해야 하나요?

A: 다음 명령어로 제거할 수 있습니다:

sudo amdgpu-uninstall

Q: ROCm 버전을 업그레이드하려면 어떻게 해야 하나요?

A: in-place 업그레이드는 지원하지 않으므로, 기존 버전을 제거하고 새 버전을 설치해야 합니다.

결론

Windows 11에서 AMD GPU를 활용한 AI 작업을 위한 ROCm 환경 구축 방법을 살펴봤습니다. WSL2와 Ubuntu 22.04를 기반으로 AMD ROCm 6.3.4와 PyTorch를 설치하고 구성하는 전체 과정을 다뤘습니다.

이 가이드를 따라 설치하면 Windows 환경에서도 AMD Radeon RX 7000 시리즈 GPU를 활용한 딥러닝 및 머신러닝 작업을 수행할 수 있습니다. ComfyUI와 같은 AI 애플리케이션도 실행할 수 있으니, 다양한 AI 프로젝트에 활용해보세요.

더 자세한 내용은 Install Radeon software for WSL with ROCm과 Install PyTorch for WSL 공식 문서를 참고하세요.