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系列和一些较新的系列。
辰迅云「云服务器」，即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘，价格低至29元/月。点击查看>>
云服务器
物理服务器
虚拟主机
跨境电商服务器