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