はじめに
以前はIntel GPUでPyTorchを使う場合には「Intel Extension for PyTorch」を使う必要がありました。touch-sp.hatenablog.com
最近PyTorchがIntel GPU をサポートし始めたようです。
pytorch.org
PyTorchのインストール
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/test/xpu
「test」と書かれていて公式にはまだリリースされていないようです。
事前準備
こちらの通りに実行すると問題なく準備は完了しました。www.intel.com