A카드 유저들은 그림을 그리거나, 비약을 만들거나, 대형 모델을 운영하거나, 머신러닝, 딥러닝까지 해보면 운이 좋을 거에요~
注意:ROCm目前仅限在Linux系统下可用,Windows暂不支持
1. ROCm 설치
RX6000 시리즈 이하 그래픽 카드는 ROCm 5.4.2 안정 버전 명령을 사용합니다.
[데스크탑 AMD Radeon RX6950XT, RX6900XT, RX6800XT, RX6800, RX6750XT, RX6750GRE, RX6700XT, RX6700GRE, RX6700, RX6650XT, RX6600XT, RX6600, RX6500XT, RX6400 및 기타 모바일 AMD Radeon RX 6을 지원합니다. 850M XT, RX6800M, RX6700M, RX6650M XT,
RX6800S , RX6700S, RX6650M, RX6600S, RX6600M, RX6500M, RX6300M 등]
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.4.2
최신 RX7000 시리즈 그래픽 카드에는 ROCm 버전 5.6이 필요합니다( --pre
최신 시험판 또는 개발 버전을 쿼리하는 옵션도 포함).
[데스크톱 AMD Radeon RX7900 XTX, RX7900 XT, RX7800 XT, RX7700 XT, Radeon RX7600 등을 지원합니다.
모바일 RX 7600M XT 등을 지원합니다.]
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm5.6
2. 설치가 완료된 후 Python 환경에서 cuda를 사용할 수 있는지 확인합니다.
import torch
torch.cuda.is_available()
코드 실행 결과 출력 true
이 OK입니다.
3. 프로젝트를 실행하기 전에 필요한 환경 변수 구성이 있습니다.
3.1 먼저 Linux 구성 파일을 엽니다.
vim ~/.bashrc
3.2 끝에 줄을 추가합니다.
그래픽 카드가 RX6000 시리즈 이하인 경우 다음을 복사하여 붙여넣으세요.
HSA_OVERRIDE_GFX_VERSION=10.3.0 내보내기
RX7000 시리즈 그래픽 카드인 경우 다음을 복사하여 붙여넣으세요.
HSA_OVERRIDE_GFX_VERSION=11.0.0 내보내기
3.3 구성을 저장하고 종료하여 구성이 즉시 적용되도록 합니다.
source ~/.bashrc
완전한 텍스트