DirectML을 사용하는 PyTorch를 사용하면 DirectX 12 지원 GPU에 대한 학습 및 유추가 가능합니다. DirectML을 사용한 PyTorch는 공개 미리 보기로 제공되며 Windows 10 버전 1709부터 네이티브 Windows 작동합니다.
Windows 버전 확인
Windows 버전 및 빌드 번호를 확인하려면 Windows 로고 키 + R을 선택하고, 입력winver하고, 확인을 선택합니다. 빌드가 오래된 경우 Windows 10 버전 1709 이상으로 업데이트합니다.
GPU 드라이버 업데이트 확인
Windows 업데이트 또는 하드웨어 제조업체의 웹 사이트를 통해 GPU에 사용할 수 있는 최신 드라이버를 설치합니다.
Python 설정
Python 환경을 설치합니다. Miniconda를 사용하는 경우 아키텍처에 대한 Windows 설치 관리자를 다운로드하여 실행합니다.
그런 다음, 다음과 같은 환경을 pytorch-directml만들고 활성화합니다.
conda create --name pytorch-directml python=3.10
conda activate pytorch-directml
DirectML을 사용하여 PyTorch 설치
torch-directml 패키지를 설치합니다.
pip install torch-directml
설치 확인
Python 시작하고 다음 코드를 실행하여 DirectML 디바이스에 두 개의 텐서를 추가합니다.
import torch
import torch_directml
dml = torch_directml.device()
tensor1 = torch.tensor([1]).to(dml)
tensor2 = torch.tensor([2]).to(dml)
dml_algebra = tensor1 + tensor2
print(dml_algebra.item())
예상 출력:
3
샘플 및 피드백
예제는 DirectML PyTorch 샘플을 참조하세요. 패키지 문제를 보고하거나 기능을 요청하려면 DirectML 문제 추적기를 사용합니다.