在 Windows 及 Linux 分別安裝如下套件

#Windows
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu130 --no-cache-dir

#Linux
pip install torch torchvision torchaudio

batch

每次 batch 不要超過 1e8(1億)，否則 12 G 的 VRAM 也不夠用。

使用 CPU

底下程式在主記憶体(RAM, 非顯卡記憶体)配置 x, y 變數，然後使用 CPU 計算 batch = 1 億，每個 epoch 約需 0.9 秒多。若是使用 numpy，每個 epoch 約需 2 秒。