PyTorch是一個基于Python的科學計算庫,廣泛應用于深度學習任務。PyTorch支持的GPU主要取決于你的系統上安裝的CUDA版本。以下是關于PyTorch支持的GPU的信息:
PyTorch支持的CUDA版本會隨著PyTorch版本的更新而變化。例如,安裝最新版的PyTorch可能需要CUDA 12.1或更高版本。
雖然PyTorch的具體支持GPU型號沒有直接列出,但是可以通過支持的CUDA版本來間接推斷。例如,CUDA 12.2支持的顯卡驅動版本為535.171.04,這通常包括NVIDIA的GeForce GTX系列和一些較新的系列。