温馨提示×

Linux系统支持哪些PyTorch版本

小樊
49
2025-09-10 19:52:07
栏目: 智能运维

Linux系统支持大多数PyTorch版本,具体取决于系统配置,尤其是Python和CUDA版本。以下是详细说明:

  • Python版本要求:通常支持Python 3.7及以上版本,推荐使用Python 3.8/3.9。
  • CUDA版本对应关系
    • PyTorch 2.1.x支持CUDA 11.8、12.1,推荐12.1。
    • PyTorch 2.0.x支持CUDA 11.7、11.8,推荐11.8。
    • PyTorch 1.13.x支持CUDA 11.6、11.7,推荐11.7。
  • 发行版支持:官方推荐Ubuntu 16.04/18.04/20.04或更高版本,CentOS 7+、RHEL 7+等主流发行版也通常支持。

0