pyg-lib
安装
我们为所有主要的操作系统/PyTorch/CUDA组合提供预编译的Python轮子,支持Python 3.8到3.12版本,详情请参见此处。
请注意,目前不支持Windows轮子(我们正在努力尽快修复这个问题)。
要安装轮子,只需运行以下命令:
pip install pyg-lib -f https://data.pyg.org/whl/torch-${TORCH}+${CUDA}.html
其中:
${TORCH}
应替换为 1.12.0
、1.13.0
、2.0.0
、2.1.0
、2.2.0
、2.3.0
或 2.4.0
中的一个
${CUDA}
应替换为 cpu
、cu102
、cu113
、cu116
、cu117
、cu118
或 cu121
中的一个
支持以下组合:
PyTorch 2.4 | cpu | cu113 | cu116 | cu117 | cu118 | cu121 | cu124 |
---|
Linux | ✅ | | | | ✅ | ✅ | ✅ |
Windows | ✅ | | | | ✅ | ✅ | ✅ |
macOS | ✅ | | | | | | |
PyTorch 2.3 | cpu | cu113 | cu116 | cu117 | cu118 | cu121 | cu124 |
---|
Linux | ✅ | | | | ✅ | ✅ | |
Windows | ✅ | | | | ✅ | ✅ | |
macOS | ✅ | | | | | | |
PyTorch 2.2 | cpu | cu113 | cu116 | cu117 | cu118 | cu121 | cu124 |
---|
Linux | ✅ | | | | ✅ | ✅ | |
Windows | ✅ | | | | ✅ | ✅ | |
macOS | ✅ | | | | | | |
PyTorch 2.1 | cpu | cu113 | cu116 | cu117 | cu118 | cu121 | cu124 |
---|
Linux | ✅ | | | | ✅ | ✅ | |
Windows | ✅ | | | | ✅ | ✅ | |
macOS | ✅ | | | | | | |
PyTorch 2.0 | cpu | cu113 | cu116 | cu117 | cu118 | cu121 | cu124 |
---|
Linux | ✅ | | | ✅ | ✅ | ✅ | |
Windows | ✅ | | | ✅ | ✅ | | |
macOS | ✅ | | | | | | |
PyTorch 1.13 | cpu | cu113 | cu116 | cu117 | cu118 | cu121 | cu124 |
---|
Linux | ✅ | | ✅ | ✅ | | | |
Windows | ✅ | | ✅ | ✅ | | | |
macOS | ✅ | | | | | | |
PyTorch 1.12 | cpu | cu113 | cu116 | cu117 | cu118 | cu121 | cu124 |
---|
Linux | ✅ | ✅ | ✅ | | | | |
Windows | ✅ | ✅ | ✅ | | | | |
macOS | ✅ | | | | | | |
从每日构建版本安装
为Linux提供从Python 3.8到3.12版本的每日构建轮子:
pip install pyg-lib -f https://data.pyg.org/whl/nightly/torch-${TORCH}+${CUDA}.html
从主分支安装
pip install ninja wheel
pip install git+https://github.com/pyg-team/pyg-lib.git