Ubuntu 25.04에서 AMD ROCm 6.4.1 및 AI Max 395+ GPU 환경 구축 완벽 가이드

Ubuntu 25.04에서 AMD ROCm 6.4.1 및 AI Max 395+ GPU 환경 구축 완벽 가이드

최신 AMD AI Max 395+ GPU(gfx1151)와 ROCm을 Ubuntu 25.04에서 성공적으로 설치한 실제 경험을 바탕으로, 설치부터 환경 점검까지 모든 과정을 상세히 정리합니다. 공식 지원은 Ubuntu 24.04 LTS까지지만, 최신 커널 환경에서도 딥러닝 및 GPU 가속을 활용하고 싶은 분께 도움이 될 것입니다.

1. 시스템 준비

1-1. 시스템 업데이트

 

bash

sudo apt update sudo apt upgrade -y sudo apt install -y wget gnupg lsb-release

1-2. ROCm 저장소 및 GPG 키 등록

ROCm 저장소는 Ubuntu 24.04(noble)용을 사용합니다.

 

bash

sudo mkdir -p /etc/apt/keyrings wget https://repo.radeon.com/rocm/rocm.gpg.key -O - | gpg --dearmor | sudo tee /etc/apt/keyrings/rocm.gpg > /dev/null sudo chmod 644 /etc/apt/keyrings/rocm.gpg echo "deb [arch=amd64 signed-by=/etc/apt/keyrings/rocm.gpg] https://repo.radeon.com/rocm/apt/6.4.1 noble main" | sudo tee /etc/apt/sources.list.d/rocm.list echo -e 'Package: *\nPin: release o=repo.radeon.com\nPin-Priority: 600' | sudo tee /etc/apt/preferences.d/rocm-pin-600 sudo apt update

2. AMDGPU Installer로 ROCm 및 그래픽 드라이버 설치

2-1. AMDGPU Installer 다운로드 및 설치

 

bash

wget https://repo.radeon.com/amdgpu-install/6.4.1/ubuntu/noble/amdgpu-install_6.4.60401-1_all.deb sudo apt install ./amdgpu-install_6.4.60401-1_all.deb sudo apt update

2-2. DKMS 없이 ROCm 및 그래픽 라이브러리 설치

최신 커널에서는 DKMS 모듈 빌드가 실패할 수 있으므로, 아래와 같이 설치합니다.

 

bash

sudo amdgpu-install --usecase=rocm,graphics --no-dkms

설치 중 프롬프트가 나오면 안내에 따라 진행하세요.

2-3. 권한 추가 및 재부팅

 

bash

sudo usermod -a -G render,video $USER sudo reboot

3. 설치 확인 및 환경 점검

3-1. ROCm 및 GPU 인식 확인

 

bash

/opt/rocm/bin/rocminfo

• gfx1151 GPU, ROCk module loaded 등 메시지가 정상적으로 나오면 성공입니다.

3-2. OpenCL 런타임 및 유틸리티 설치

OpenCL 런타임이 이미 설치되어 있어도, clinfo 유틸리티는 별도 설치가 필요합니다.

 

bash

sudo apt install clinfo clinfo

• AMD GPU가 OpenCL 디바이스로 인식되는지 확인합니다.

3-3. 환경 변수 및 ICD 파일 확인

OpenCL이 인식되지 않을 경우:

 

bash

echo /opt/rocm/lib/libamdocl64.so | sudo tee /etc/OpenCL/vendors/amdocl64.icd export LD_LIBRARY_PATH=/opt/rocm/lib:/opt/rocm/opencl/lib clinfo

4. PyTorch 및 AI 프레임워크 설치

4-1. PyTorch ROCm 패키지 설치

 

bash

pip3 install --upgrade pip pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.4/

4-2. GPU 연산 테스트

 

python

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

• True와 함께 GPU 이름이 출력되면 성공입니다.

5. 문제 해결 팁

• DKMS 모듈이 필요 없는 최신 커널(25.04)에서는 --no-dkms 옵션이 매우 유용합니다.
• OpenCL 유틸리티가 누락된 경우, Ubuntu 저장소의 clinfo를 사용하세요.
• 환경변수 및 ICD 파일 경로를 확인하면 대부분의 인식 문제를 해결할 수 있습니다.
• 공식 지원은 24.04 LTS까지이므로, 완벽한 안정성이 필요하다면 LTS 버전을 권장합니다.

6. 결론

Ubuntu 25.04에서도 최신 AMD ROCm 및 AI Max 395+ GPU 환경을 충분히 구축할 수 있습니다. DKMS 없이 설치하면 커널 이슈를 피해 최신 환경에서도 GPU 가속을 활용할 수 있으니, 위 가이드를 따라 도전해보세요!