macOS安装Pixray遇torch==1.9.0+cu102版本匹配失败求助
macOS下Pixray安装Torch依赖报错的解决办法
问题原因
你碰到的「找不到匹配的torch==1.9.0+cu102版本」错误,核心原因是cu102是NVIDIA CUDA 10.2专属的Torch版本,而macOS不管是Intel还是M系列芯片,都不支持这个特定的CUDA包:
- Intel版macOS的CUDA支持非常有限,Torch官方提供的对应包都是CPU版本,不带
+cu102后缀; - Apple Silicon(M系列)完全不兼容CUDA,只能用适配Metal的Torch版本或者CPU版。
正确处理方式:改requirements.txt里的库版本
不用动Python 3.8.10的版本(这个版本和Pixray兼容没问题),只需要调整requirements.txt里的Torch配置:
- 打开仓库目录下的
requirements.txt文件; - 找到
torch==1.9.0+cu102这一行,换成适合你macOS芯片的版本:- Intel芯片:直接改成
torch==1.9.0(CPU版); - M系列芯片:换成适配的版本,比如
torch==1.13.1,同时要保证对应的torchvision版本和它匹配(比如Torch 1.13.1对应torchvision 0.14.1);
- Intel芯片:直接改成
- 保存文件后,重新执行
pip install -r requirements.txt。
额外提示
- 如果是M系列芯片,创建pyenv虚拟环境时,建议给Python 3.8.10打个补丁来适配ARM架构:
pyenv install 3.8.10 --patch <(curl -s https://github.com/python/cpython/pull/26555.patch); - 安装完成后,可以在Python命令行里验证Torch是否正常加载:输入
import torch; print(torch.backends.mps.is_available())(M系列)或者print(torch.cuda.is_available())(Intel),前者返回True、后者返回False都是正常情况。
内容的提问来源于stack exchange,提问作者x89
相关产品推荐
相关产品推荐

