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+等主流发行版也通常支持。