环境要求
本文档以安装环境openEuler 24.03 LTS SP3 + PyTorch 2.11.0 + CUDA 13.0为例。除此之外，本章节介绍了鲲鹏服务器上安装PyTorch可支持的其他环境信息。
可支持的版本
OS版本 |
Python版本 |
支持的PyTorch版本范围 |
|---|---|---|
openEuler 24.03 LTS SP3 |
Python 3.11.6 |
PyTorch 2.11.x；建议PyTorch 2.11.0 |
openEuler 22.03 LTS SP4 |
Python 3.9.9 |
PyTorch<=2.8.x；建议PyTorch 2.8.0 |
PyTorch 2.11.0要求Python为3.10版本及以上。openEuler 22.03 LTS SP4默认Python 3.9.9不满足该版本要求，建议在openEuler 22.03 LTS SP4上使用支持Python 3.9的PyTorch 2.8.0。
依赖组件要求
组件 |
版本 |
|---|---|
python3 |
3.11.6-20.oe2403sp3 |
python3-pip |
23.3.1-7.oe2403sp3 |
pip |
26.1.1 |
setuptools |
82.0.1 |
wheel |
0.47.0 |
PyTorch |
2.11.0 |
CUDA Toolkit |
13.0 |
使用GPU能力时，系统需要提供可访问的NVIDIA GPU设备和与CUDA 13.0匹配的NVIDIA驱动用户态库。安装验证时必须确认PyTorch为CUDA版本，并执行GPU张量计算。
父主题： 安装前准备