PyTorch 就是一台可以啟動 GPU 進行數學運算的計算機。PyTorch 於 2017 年初由 Facebook 開發的開源深度學習框架，語法簡潔優雅，概念直觀，容易上手，再加上烏俄戰爭無人機廣泛使用 Yolo 進行視覺判辨，遂漸瓜分 Tensorflow 的市場。

要讓 PyTorch 啟動 GPU，需要 nVidia 的 Cuda 及 cudnn 套件，但我們並不需要刻意在系統中安裝 Cuda 及 cudnn，因為在安裝 PyTorch 時，就會自帶安裝最適合的 Cuda/cudnn 到虛擬環境中，系統有沒有安裝 Cuda/cudnn 並不重要。不過為了在 Linux 下啟動 tensorflow GPU，還是建議手動裝一下 Cuda/cudnn

Ubuntu 2404 安裝 PyTorch

在 ubuntu 中，只要裝好顯卡驅動程式，系統不需安裝 Cuda/cudnn。安裝 PyTorch 時就會選取最高的 cu 版本，安裝指令如下如下

pip install torch torchvision torchaudio

Windows 安裝 PyTorch

在 Windows 下安裝 PyTorch 需指定 –index-url 參數，並指定 cu (cuda)版本，如下所示

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

最新可用的 cu 版本請到官網查看 https://download.pytorch.org/whl/

如果電腦的記憶体只有 16 G 以下，又同時開啟二個專案以上，安裝上面的 Pytorch 時會出現 Memory error例外，甚至當機連滑鼠鍵盤都不會動，加上 “–no-cache-dir” 即可解決。

測試是否有安裝成功，可使用如下指令，如果印出 “True”，表示安裝成功。如果沒有 nVidia 顯卡，就算 torch 安裝 cu 版本，結果還是 False。

import torch
print(torch.cuda.is_available())
結果:
True

基本格式

numpy 的基本格式是 np.array，中文稱”陣列”，資料型態可以為 np.int8, np.int32, np.float32…。