728x90
출처
- Install PyTorch and ComfyUI on ROCm 7.10.0 — AMD ROCm 7.10.0 preview
- Installation via AMDGPU installer — ROCm installation (Linux)
리눅스 버전 확인
bluesanta@bluesanta-AI-Series:~$ lsb_release -a
No LSB modules are available.
Distributor ID: Ubuntu
Description: Ubuntu 24.04.3 LTS
Release: 24.04
Codename: noble
AMD GPU 드라이버 설치
AMD GPU 설치 프로그램 설치
bluesanta@bluesanta-AI-Series:~$ wget https://repo.radeon.com/amdgpu-install/30.20.1/ubuntu/noble/amdgpu-install_7.1.1.70101-1_all.deb
bluesanta@bluesanta-AI-Series:~$ sudo dpkg -i amdgpu-install_7.1.1.70101-1_all.deb
[sudo] password for bluesanta:
Selecting previously unselected package amdgpu-install.
(Reading database ... 215673 files and directories currently installed.)
Preparing to unpack amdgpu-install_7.1.1.70101-1_all.deb ...
Unpacking amdgpu-install (30.20.1.0.30200100-2255209.24.04) ...
Setting up amdgpu-install (30.20.1.0.30200100-2255209.24.04) ...
ROCm 패키지 설치
bluesanta@bluesanta-AI-Series:~$ sudo amdgpu-install --usecase=rocm,hiplibsdk --no-dkms
설치 확인
(.venv) bluesanta@bluesanta-AI-Series:~/Application/stable_diffusion$ lsmod | grep amdgpu
amdgpu 19836928 6
amddrm_ttm_helper 12288 1 amdgpu
amdttm 131072 2 amdgpu,amddrm_ttm_helper
amddrm_buddy 24576 1 amdgpu
amdxcp 16384 1 amdgpu
amddrm_exec 12288 1 amdgpu
drm_suballoc_helper 20480 1 amdgpu
amd_sched 61440 1 amdgpu
amdkcl 36864 4 amd_sched,amdttm,amddrm_exec,amdgpu
drm_panel_backlight_quirks 12288 1 amdgpu
drm_display_helper 278528 1 amdgpu
cec 94208 2 drm_display_helper,amdgpu
i2c_algo_bit 16384 1 amdgpu
drm_ttm_helper 16384 1 amdgpu
video 77824 1 amdgpu
(.venv) bluesanta@bluesanta-AI-Series:~/Application/stable_diffusion$ dkms status
amdgpu/6.16.6-2255209.24.04, 6.14.0-37-generic, x86_64: installed
아키텍처 강제 인식
HX 370(gfx1150)은 최신 아키텍처이므로, PyTorch가 이를 인식하도록 환경 변수를 선언해야 합니다. 터미널에서 다음을 입력하거나 .bashrc에 추가
bluesanta@bluesanta-AI-Series:~$ vi ~/.bashrc
.bashrc 파일에 HSA_OVERRIDE_GFX_VERSION 값 설정
export HSA_OVERRIDE_GFX_VERSION=11.5.0
PyTorch 설치
가상환경만들기
bluesanta@bluesanta-AI-Series:~/Application/stable_diffusion$ sudo apt install python3-venv
bluesanta@bluesanta-AI-Series:~/Application/stable_diffusion$ python3 -m venv .venv
bluesanta@bluesanta-AI-Series:~/Application/stable_diffusion$ source .venv/bin/activate
(.venv) bluesanta@bluesanta-AI-Series:~/Application/stable_diffusion$
ROCm 지원 PyTorch 빌드 및 설치
(.venv) bluesanta@bluesanta-AI-Series:~/Application/stable_diffusion$ python -m pip install --index-url https://repo.amd.com/rocm/whl/gfx1150/ torch torchvision torchaudio
적용 확인
(.venv) bluesanta@bluesanta-AI-Series:~/Application/stable_diffusion$ python -c "import torch; print(torch.cuda.is_available())"
True728x90