修复PyTorch与IPEX兼容性问题 | AI生成和翻译
你遇到了两个问题:
-
版本不匹配:Intel Extension for PyTorch (IPEX) 需要 PyTorch 2.8.*,但你安装的是 PyTorch 2.7.1+cu126(CUDA版本)
-
IPEX中的错误:IPEX库本身存在一个bug - 它试图调用不存在的
os.exit(127)。正确的函数应该是sys.exit(127)
以下是解决方法:
解决方案1:安装兼容的PyTorch版本
# 卸载当前的PyTorch
pip uninstall torch torchvision torchaudio
# 安装PyTorch 2.8.*(CPU版本,因为你使用的是Intel GPU)
pip install torch==2.8.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
# 重新安装IPEX
pip install intel-extension-for-pytorch
解决方案2:升级到最新版本
# 将PyTorch和IPEX都升级到最新的兼容版本
pip install --upgrade torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
pip install --upgrade intel-extension-for-pytorch
解决方案3:如果你需要CUDA版本的PyTorch
如果你确实需要CUDA支持(用于NVIDIA GPU),那么你根本不应该使用Intel Extension for PyTorch - 它们服务于不同的硬件:
- Intel GPU/CPU:使用PyTorch CPU构建 + IPEX
- NVIDIA GPU:使用PyTorch CUDA构建(不需要IPEX)
关于该错误的说明
os.exit()的错误表明你可能使用的是过时版本的IPEX。最新版本应该已经修复了这个问题。在解决PyTorch版本不匹配问题后,请确保安装最新版本的IPEX。
你实际想要使用什么硬件 - Intel GPU、Intel CPU还是NVIDIA GPU?