PyTorch が Intel GPU のサポートを開始したようです。

はじめに

以前は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

最後に

PyTorchを使う人にとって今なら4万円台でVRAM 16GBを手に入れることができます。